Replaced old ray image with new 2.20.0 image

Bobbins228 · Bobbins228 · commit d6cf2cbf90a1 · 2024-06-18T10:11:05.000+01:00
diff --git a/src/codeflare_sdk/templates/base-template.yaml b/src/codeflare_sdk/templates/base-template.yaml
@@ -78,7 +78,7 @@ spec:
               containers:
               # The Ray head pod
               - name: ray-head
-                image: quay.io/project-codeflare/ray:latest-py39-cu118
+                image: quay.io/project-codeflare/ray:2.20.0-py39-cu118
                 imagePullPolicy: Always
                 ports:
                 - containerPort: 6379
@@ -161,7 +161,7 @@ spec:
             spec:
               containers:
               - name: machine-learning # must consist of lower case alphanumeric characters or '-', and must start and end with an alphanumeric character (e.g. 'my-name',  or '123-abc'
-                image: quay.io/project-codeflare/ray:latest-py39-cu118
+                image: quay.io/project-codeflare/ray:2.20.0-py39-cu118
                 # environment variables to set in the container.Optional.
                 # Refer to https://kubernetes.io/docs/tasks/inject-data-application/define-environment-variable-container/
                 lifecycle:
diff --git a/tests/e2e/support.py b/tests/e2e/support.py
@@ -8,7 +8,7 @@
 
 
 def get_ray_image():
-    default_ray_image = "quay.io/project-codeflare/ray:latest-py39-cu118"
+    default_ray_image = "quay.io/project-codeflare/ray:2.20.0-py39-cu118"
     return os.getenv("RAY_IMAGE", default_ray_image)
 
 
diff --git a/tests/test-case-bad.yaml b/tests/test-case-bad.yaml
@@ -42,7 +42,7 @@ spec:
                   valueFrom:
                     fieldRef:
                       fieldPath: status.podIP
-                image: quay.io/project-codeflare/ray:latest-py39-cu118
+                image: quay.io/project-codeflare/ray:2.20.0-py39-cu118
                 imagePullPolicy: Always
                 lifecycle:
                   preStop:
@@ -90,7 +90,7 @@ spec:
                   valueFrom:
                     fieldRef:
                       fieldPath: status.podIP
-                image: quay.io/project-codeflare/ray:latest-py39-cu118
+                image: quay.io/project-codeflare/ray:2.20.0-py39-cu118
                 lifecycle:
                   preStop:
                     exec:
diff --git a/tests/test-case-no-mcad.yamls b/tests/test-case-no-mcad.yamls
@@ -34,7 +34,7 @@ spec:
     template:
       spec:
         containers:
-        - image: quay.io/project-codeflare/ray:latest-py39-cu118
+        - image: quay.io/project-codeflare/ray:2.20.0-py39-cu118
           imagePullPolicy: Always
           lifecycle:
             preStop:
@@ -107,7 +107,7 @@ spec:
           key: value
       spec:
         containers:
-        - image: quay.io/project-codeflare/ray:latest-py39-cu118
+        - image: quay.io/project-codeflare/ray:2.20.0-py39-cu118
           lifecycle:
             preStop:
               exec:
diff --git a/tests/test-case.yaml b/tests/test-case.yaml
@@ -38,7 +38,7 @@ spec:
           template:
             spec:
               containers:
-              - image: quay.io/project-codeflare/ray:latest-py39-cu118
+              - image: quay.io/project-codeflare/ray:2.20.0-py39-cu118
                 imagePullPolicy: Always
                 lifecycle:
                   preStop:
@@ -111,7 +111,7 @@ spec:
                 key: value
             spec:
               containers:
-              - image: quay.io/project-codeflare/ray:latest-py39-cu118
+              - image: quay.io/project-codeflare/ray:2.20.0-py39-cu118
                 lifecycle:
                   preStop:
                     exec:
diff --git a/tests/test-default-appwrapper.yaml b/tests/test-default-appwrapper.yaml
@@ -38,7 +38,7 @@ spec:
           template:
             spec:
               containers:
-              - image: quay.io/project-codeflare/ray:latest-py39-cu118
+              - image: quay.io/project-codeflare/ray:2.20.0-py39-cu118
                 imagePullPolicy: Always
                 lifecycle:
                   preStop:
@@ -110,7 +110,7 @@ spec:
                 key: value
             spec:
               containers:
-              - image: quay.io/project-codeflare/ray:latest-py39-cu118
+              - image: quay.io/project-codeflare/ray:2.20.0-py39-cu118
                 lifecycle:
                   preStop:
                     exec:
diff --git a/tests/unit_test.py b/tests/unit_test.py
@@ -263,7 +263,7 @@ def test_config_creation():
     assert config.min_cpus == 3 and config.max_cpus == 4
     assert config.min_memory == "5G" and config.max_memory == "6G"
     assert config.num_gpus == 7
-    assert config.image == "quay.io/project-codeflare/ray:latest-py39-cu118"
+    assert config.image == "quay.io/project-codeflare/ray:2.20.0-py39-cu118"
     assert config.template == f"{parent}/src/codeflare_sdk/templates/base-template.yaml"
     assert config.machine_types == ["cpu.small", "gpu.large"]
     assert config.image_pull_secrets == ["unit-test-pull-secret"]
@@ -402,7 +402,7 @@ def test_cluster_creation_no_mcad_local_queue(mocker):
         num_gpus=7,
         machine_types=["cpu.small", "gpu.large"],
         image_pull_secrets=["unit-test-pull-secret"],
-        image="quay.io/project-codeflare/ray:latest-py39-cu118",
+        image="quay.io/project-codeflare/ray:2.20.0-py39-cu118",
         write_to_file=True,
         appwrapper=False,
         local_queue="local-queue-default",
@@ -430,7 +430,7 @@ def test_default_cluster_creation(mocker):
     )
     default_config = ClusterConfiguration(
         name="unit-test-default-cluster",
-        image="quay.io/project-codeflare/ray:latest-py39-cu118",
+        image="quay.io/project-codeflare/ray:2.20.0-py39-cu118",
         appwrapper=True,
     )
     cluster = Cluster(default_config)
@@ -779,7 +779,7 @@ def test_ray_job_wrapping(mocker):
         return_value=get_local_queue("kueue.x-k8s.io", "v1beta1", "ns", "localqueues"),
     )
     cluster = cluster = createClusterWithConfig(mocker)
-    cluster.config.image = "quay.io/project-codeflare/ray:latest-py39-cu118"
+    cluster.config.image = "quay.io/project-codeflare/ray:2.20.0-py39-cu118"
     mocker.patch(
         "ray.job_submission.JobSubmissionClient._check_connection_and_version_with_url",
         return_value="None",
@@ -899,7 +899,7 @@ def test_ray_details(mocker, capsys):
         ClusterConfiguration(
             name="raytest2",
             namespace="ns",
-            image="quay.io/project-codeflare/ray:latest-py39-cu118",
+            image="quay.io/project-codeflare/ray:2.20.0-py39-cu118",
             write_to_file=True,
             appwrapper=True,
             local_queue="local_default_queue",
@@ -2292,7 +2292,7 @@ def test_cluster_status(mocker):
         ClusterConfiguration(
             name="test",
             namespace="ns",
-            image="quay.io/project-codeflare/ray:latest-py39-cu118",
+            image="quay.io/project-codeflare/ray:2.20.0-py39-cu118",
             write_to_file=True,
             appwrapper=True,
             local_queue="local_default_queue",
@@ -2387,7 +2387,7 @@ def test_wait_ready(mocker, capsys):
         ClusterConfiguration(
             name="test",
             namespace="ns",
-            image="quay.io/project-codeflare/ray:latest-py39-cu118",
+            image="quay.io/project-codeflare/ray:2.20.0-py39-cu118",
             write_to_file=True,
             appwrapper=True,
             local_queue="local-queue-default",
@@ -2614,7 +2614,7 @@ def throw_if_getting_raycluster(group, version, namespace, plural):
     cluster = Cluster(
         ClusterConfiguration(
             "test_cluster",
-            image="quay.io/project-codeflare/ray:latest-py39-cu118",
+            image="quay.io/project-codeflare/ray:2.20.0-py39-cu118",
             write_to_file=False,
         )
     )
diff --git a/tests/unit_test_support.py b/tests/unit_test_support.py
@@ -17,7 +17,7 @@ def createClusterConfig():
         appwrapper=True,
         machine_types=["cpu.small", "gpu.large"],
         image_pull_secrets=["unit-test-pull-secret"],
-        image="quay.io/project-codeflare/ray:latest-py39-cu118",
+        image="quay.io/project-codeflare/ray:2.20.0-py39-cu118",
         write_to_file=True,
     )
     return config

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@ def createClusterConfig():`
`17`	`17`	`appwrapper=True,`
`18`	`18`	`machine_types=["cpu.small", "gpu.large"],`
`19`	`19`	`image_pull_secrets=["unit-test-pull-secret"],`
`20`		`- image="quay.io/project-codeflare/ray:latest-py39-cu118",`
	`20`	`+ image="quay.io/project-codeflare/ray:2.20.0-py39-cu118",`
`21`	`21`	`write_to_file=True,`
`22`	`22`	`)`
`23`	`23`	`return config`