Skip to content

Commit

Permalink
Simplify smoke test examples resource consumption (#25)
Browse files Browse the repository at this point in the history
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.
  • Loading branch information
maxmynter authored May 3, 2024
1 parent de1a4cc commit e1ebc32
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion example_hello.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
name="localhost:5000/hello-world-dev",
tag="latest",
),
resources=ResourceOptions(memory="4Gi", cpu="2"),
resources=ResourceOptions(memory="1Gi", cpu="1"),
scheduling=SchedulingOptions(
priority_class="background", queue_name="user-queue"
),
Expand Down
4 changes: 2 additions & 2 deletions example_mnist_tf.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,9 @@
@job(
options=JobOptions(
image=ImageOptions(
spec=Path("example-docker.yaml"), name="mlops:5000/tf-example"
spec=Path("example-docker.yaml"), name="localhost:5000/tf-example"
),
resources=ResourceOptions(memory="4Gi", cpu="2", gpu=1 if USE_GPU else None),
resources=ResourceOptions(memory="1Gi", cpu="2", gpu=1 if USE_GPU else None),
)
)
def mnist_train() -> None:
Expand Down

0 comments on commit e1ebc32

Please sign in to comment.