Skip to content

Commit 74d8846

Browse files
carsonmhMaxusmusti
authored andcommitted
Fix: unit test failing because of ray cluster obj changed (#208)
1 parent 89ac249 commit 74d8846

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/unit_test.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -793,10 +793,10 @@ def get_ray_obj(group, version, namespace, plural, cls=None):
793793
"workerGroupSpecs": [
794794
{
795795
"groupName": "small-group-quicktest",
796-
"maxReplicas": 2,
797-
"minReplicas": 2,
796+
"maxReplicas": 1,
797+
"minReplicas": 1,
798798
"rayStartParams": {"block": "true", "num-gpus": "0"},
799-
"replicas": 2,
799+
"replicas": 1,
800800
"template": {
801801
"metadata": {
802802
"annotations": {"key": "value"},
@@ -1552,7 +1552,7 @@ def test_get_cluster(mocker):
15521552
cluster_config.image
15531553
== "ghcr.io/foundation-model-stack/base:ray2.1.0-py38-gpu-pytorch1.12.0cu116-20221213-193103"
15541554
)
1555-
assert cluster_config.min_worker == 2 and cluster_config.max_worker == 2
1555+
assert cluster_config.min_worker == 1 and cluster_config.max_worker == 1
15561556

15571557

15581558
def test_list_clusters(mocker, capsys):

0 commit comments

Comments
 (0)