Skip to content

Commit e1ebc32

Browse files
authored
Simplify smoke test examples resource consumption (#25)
such that it puts less strain on developer machines. Also change the namespace from 'mlops' back to 'localhost' to avoid manual namespace setup in a new cluster instance upon testing. If more resources or a custom namespace is required, we can set it up quickly when needed. Otherwise, for the default, this setup will be more convenient.
1 parent de1a4cc commit e1ebc32

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

example_hello.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
name="localhost:5000/hello-world-dev",
1212
tag="latest",
1313
),
14-
resources=ResourceOptions(memory="4Gi", cpu="2"),
14+
resources=ResourceOptions(memory="1Gi", cpu="1"),
1515
scheduling=SchedulingOptions(
1616
priority_class="background", queue_name="user-queue"
1717
),

example_mnist_tf.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,9 @@
88
@job(
99
options=JobOptions(
1010
image=ImageOptions(
11-
spec=Path("example-docker.yaml"), name="mlops:5000/tf-example"
11+
spec=Path("example-docker.yaml"), name="localhost:5000/tf-example"
1212
),
13-
resources=ResourceOptions(memory="4Gi", cpu="2", gpu=1 if USE_GPU else None),
13+
resources=ResourceOptions(memory="1Gi", cpu="2", gpu=1 if USE_GPU else None),
1414
)
1515
)
1616
def mnist_train() -> None:

0 commit comments

Comments
 (0)