@@ -261,7 +261,7 @@ def test_config_creation():
261
261
assert config .min_cpus == 3 and config .max_cpus == 4
262
262
assert config .min_memory == "5G" and config .max_memory == "6G"
263
263
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"
265
265
assert config .template == f"{ parent } /src/codeflare_sdk/templates/base-template.yaml"
266
266
assert config .machine_types == ["cpu.small" , "gpu.large" ]
267
267
assert config .image_pull_secrets == ["unit-test-pull-secret" ]
@@ -400,7 +400,7 @@ def test_cluster_creation_no_mcad_local_queue(mocker):
400
400
num_gpus = 7 ,
401
401
machine_types = ["cpu.small" , "gpu.large" ],
402
402
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" ,
404
404
write_to_file = True ,
405
405
appwrapper = False ,
406
406
local_queue = "local-queue-default" ,
@@ -428,7 +428,7 @@ def test_default_cluster_creation(mocker):
428
428
)
429
429
default_config = ClusterConfiguration (
430
430
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" ,
432
432
appwrapper = True ,
433
433
)
434
434
cluster = Cluster (default_config )
@@ -493,7 +493,7 @@ def test_cluster_with_custom_volumes(mocker):
493
493
494
494
test_config = ClusterConfiguration (
495
495
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" ,
497
497
volume_mounts = volume_mounts ,
498
498
volumes = volumes ,
499
499
)
@@ -839,7 +839,7 @@ def test_ray_job_wrapping(mocker):
839
839
return_value = get_local_queue ("kueue.x-k8s.io" , "v1beta1" , "ns" , "localqueues" ),
840
840
)
841
841
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"
843
843
mocker .patch (
844
844
"ray.job_submission.JobSubmissionClient._check_connection_and_version_with_url" ,
845
845
return_value = "None" ,
@@ -959,7 +959,7 @@ def test_ray_details(mocker, capsys):
959
959
ClusterConfiguration (
960
960
name = "raytest2" ,
961
961
namespace = "ns" ,
962
- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
962
+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
963
963
write_to_file = True ,
964
964
appwrapper = True ,
965
965
local_queue = "local_default_queue" ,
@@ -2352,7 +2352,7 @@ def test_cluster_status(mocker):
2352
2352
ClusterConfiguration (
2353
2353
name = "test" ,
2354
2354
namespace = "ns" ,
2355
- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
2355
+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
2356
2356
write_to_file = True ,
2357
2357
appwrapper = True ,
2358
2358
local_queue = "local_default_queue" ,
@@ -2447,7 +2447,7 @@ def test_wait_ready(mocker, capsys):
2447
2447
ClusterConfiguration (
2448
2448
name = "test" ,
2449
2449
namespace = "ns" ,
2450
- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
2450
+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
2451
2451
write_to_file = True ,
2452
2452
appwrapper = True ,
2453
2453
local_queue = "local-queue-default" ,
@@ -2674,7 +2674,7 @@ def throw_if_getting_raycluster(group, version, namespace, plural):
2674
2674
cluster = Cluster (
2675
2675
ClusterConfiguration (
2676
2676
"test_cluster" ,
2677
- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
2677
+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
2678
2678
write_to_file = False ,
2679
2679
)
2680
2680
)
0 commit comments