@@ -261,7 +261,7 @@ def test_config_creation():
261261 assert config .min_cpus == 3 and config .max_cpus == 4
262262 assert config .min_memory == "5G" and config .max_memory == "6G"
263263 assert config .num_gpus == 7
264- assert config .image == "quay.io/project-codeflare/ray:latest -py39-cu118"
264+ assert config .image == "quay.io/project-codeflare/ray:2.20.0 -py39-cu118"
265265 assert config .template == f"{ parent } /src/codeflare_sdk/templates/base-template.yaml"
266266 assert config .machine_types == ["cpu.small" , "gpu.large" ]
267267 assert config .image_pull_secrets == ["unit-test-pull-secret" ]
@@ -400,7 +400,7 @@ def test_cluster_creation_no_mcad_local_queue(mocker):
400400 num_gpus = 7 ,
401401 machine_types = ["cpu.small" , "gpu.large" ],
402402 image_pull_secrets = ["unit-test-pull-secret" ],
403- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
403+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
404404 write_to_file = True ,
405405 appwrapper = False ,
406406 local_queue = "local-queue-default" ,
@@ -428,7 +428,7 @@ def test_default_cluster_creation(mocker):
428428 )
429429 default_config = ClusterConfiguration (
430430 name = "unit-test-default-cluster" ,
431- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
431+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
432432 appwrapper = True ,
433433 )
434434 cluster = Cluster (default_config )
@@ -493,7 +493,7 @@ def test_cluster_with_custom_volumes(mocker):
493493
494494 test_config = ClusterConfiguration (
495495 name = "unit-test-volume-cluster" ,
496- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
496+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
497497 volume_mounts = volume_mounts ,
498498 volumes = volumes ,
499499 )
@@ -839,7 +839,7 @@ def test_ray_job_wrapping(mocker):
839839 return_value = get_local_queue ("kueue.x-k8s.io" , "v1beta1" , "ns" , "localqueues" ),
840840 )
841841 cluster = cluster = createClusterWithConfig (mocker )
842- cluster .config .image = "quay.io/project-codeflare/ray:latest -py39-cu118"
842+ cluster .config .image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118"
843843 mocker .patch (
844844 "ray.job_submission.JobSubmissionClient._check_connection_and_version_with_url" ,
845845 return_value = "None" ,
@@ -959,7 +959,7 @@ def test_ray_details(mocker, capsys):
959959 ClusterConfiguration (
960960 name = "raytest2" ,
961961 namespace = "ns" ,
962- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
962+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
963963 write_to_file = True ,
964964 appwrapper = True ,
965965 local_queue = "local_default_queue" ,
@@ -2352,7 +2352,7 @@ def test_cluster_status(mocker):
23522352 ClusterConfiguration (
23532353 name = "test" ,
23542354 namespace = "ns" ,
2355- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
2355+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
23562356 write_to_file = True ,
23572357 appwrapper = True ,
23582358 local_queue = "local_default_queue" ,
@@ -2447,7 +2447,7 @@ def test_wait_ready(mocker, capsys):
24472447 ClusterConfiguration (
24482448 name = "test" ,
24492449 namespace = "ns" ,
2450- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
2450+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
24512451 write_to_file = True ,
24522452 appwrapper = True ,
24532453 local_queue = "local-queue-default" ,
@@ -2674,7 +2674,7 @@ def throw_if_getting_raycluster(group, version, namespace, plural):
26742674 cluster = Cluster (
26752675 ClusterConfiguration (
26762676 "test_cluster" ,
2677- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
2677+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
26782678 write_to_file = False ,
26792679 )
26802680 )
0 commit comments