@@ -263,7 +263,7 @@ def test_config_creation():
263263 assert config .min_cpus == 3 and config .max_cpus == 4
264264 assert config .min_memory == "5G" and config .max_memory == "6G"
265265 assert config .num_gpus == 7
266- assert config .image == "quay.io/project-codeflare/ray:latest -py39-cu118"
266+ assert config .image == "quay.io/project-codeflare/ray:2.20.0 -py39-cu118"
267267 assert config .template == f"{ parent } /src/codeflare_sdk/templates/base-template.yaml"
268268 assert config .machine_types == ["cpu.small" , "gpu.large" ]
269269 assert config .image_pull_secrets == ["unit-test-pull-secret" ]
@@ -399,7 +399,7 @@ def test_cluster_creation_no_mcad_local_queue(mocker):
399399 num_gpus = 7 ,
400400 machine_types = ["cpu.small" , "gpu.large" ],
401401 image_pull_secrets = ["unit-test-pull-secret" ],
402- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
402+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
403403 write_to_file = True ,
404404 appwrapper = False ,
405405 local_queue = "local-queue-default" ,
@@ -427,7 +427,7 @@ def test_default_cluster_creation(mocker):
427427 )
428428 default_config = ClusterConfiguration (
429429 name = "unit-test-default-cluster" ,
430- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
430+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
431431 appwrapper = True ,
432432 )
433433 cluster = Cluster (default_config )
@@ -776,7 +776,7 @@ def test_ray_job_wrapping(mocker):
776776 return_value = get_local_queue ("kueue.x-k8s.io" , "v1beta1" , "ns" , "localqueues" ),
777777 )
778778 cluster = cluster = createClusterWithConfig (mocker )
779- cluster .config .image = "quay.io/project-codeflare/ray:latest -py39-cu118"
779+ cluster .config .image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118"
780780 mocker .patch (
781781 "ray.job_submission.JobSubmissionClient._check_connection_and_version_with_url" ,
782782 return_value = "None" ,
@@ -896,7 +896,7 @@ def test_ray_details(mocker, capsys):
896896 ClusterConfiguration (
897897 name = "raytest2" ,
898898 namespace = "ns" ,
899- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
899+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
900900 write_to_file = True ,
901901 appwrapper = True ,
902902 local_queue = "local_default_queue" ,
@@ -2289,7 +2289,7 @@ def test_cluster_status(mocker):
22892289 ClusterConfiguration (
22902290 name = "test" ,
22912291 namespace = "ns" ,
2292- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
2292+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
22932293 write_to_file = True ,
22942294 appwrapper = True ,
22952295 local_queue = "local_default_queue" ,
@@ -2384,7 +2384,7 @@ def test_wait_ready(mocker, capsys):
23842384 ClusterConfiguration (
23852385 name = "test" ,
23862386 namespace = "ns" ,
2387- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
2387+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
23882388 write_to_file = True ,
23892389 appwrapper = True ,
23902390 local_queue = "local-queue-default" ,
@@ -2611,7 +2611,7 @@ def throw_if_getting_raycluster(group, version, namespace, plural):
26112611 cluster = Cluster (
26122612 ClusterConfiguration (
26132613 "test_cluster" ,
2614- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
2614+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
26152615 write_to_file = False ,
26162616 )
26172617 )
0 commit comments