Skip to content

Commit 1fa288d

Browse files
authored
Fix: unit test failing because of ray cluster obj changed (#208)
1 parent a8e9156 commit 1fa288d

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

Diff for: tests/unit_test.py

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

15581558

15591559
def test_list_clusters(mocker, capsys):

0 commit comments

Comments
 (0)