Skip to content

Commit 488284a

Browse files
committed
Simplify smoke test examples resource consumption
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 488284a

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=0 if USE_GPU else None),
1414
)
1515
)
1616
def mnist_train() -> None:

0 commit comments

Comments
 (0)