From dbb88489bf97227a033b32df2c2f25247ed070a5 Mon Sep 17 00:00:00 2001
From: carsonmh <thecarsonharrell@gmail.com>
Date: Mon, 10 Jul 2023 11:29:52 -0700
Subject: [PATCH] Fix: unit test failing because of ray cluster obj changed

---
 tests/unit_test.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/unit_test.py b/tests/unit_test.py
index 8174f0b54..3c69c4b6b 100644
--- a/tests/unit_test.py
+++ b/tests/unit_test.py
@@ -794,10 +794,10 @@ def get_ray_obj(group, version, namespace, plural, cls=None):
                     "workerGroupSpecs": [
                         {
                             "groupName": "small-group-quicktest",
-                            "maxReplicas": 2,
-                            "minReplicas": 2,
+                            "maxReplicas": 1,
+                            "minReplicas": 1,
                             "rayStartParams": {"block": "true", "num-gpus": "0"},
-                            "replicas": 2,
+                            "replicas": 1,
                             "template": {
                                 "metadata": {
                                     "annotations": {"key": "value"},
@@ -1553,7 +1553,7 @@ def test_get_cluster(mocker):
         cluster_config.image
         == "ghcr.io/foundation-model-stack/base:ray2.1.0-py38-gpu-pytorch1.12.0cu116-20221213-193103"
     )
-    assert cluster_config.min_worker == 2 and cluster_config.max_worker == 2
+    assert cluster_config.min_worker == 1 and cluster_config.max_worker == 1
 
 
 def test_list_clusters(mocker, capsys):