@@ -263,7 +263,7 @@ def test_config_creation():
263
263
assert config .min_cpus == 3 and config .max_cpus == 4
264
264
assert config .min_memory == "5G" and config .max_memory == "6G"
265
265
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"
267
267
assert config .template == f"{ parent } /src/codeflare_sdk/templates/base-template.yaml"
268
268
assert config .machine_types == ["cpu.small" , "gpu.large" ]
269
269
assert config .image_pull_secrets == ["unit-test-pull-secret" ]
@@ -402,7 +402,7 @@ def test_cluster_creation_no_mcad_local_queue(mocker):
402
402
num_gpus = 7 ,
403
403
machine_types = ["cpu.small" , "gpu.large" ],
404
404
image_pull_secrets = ["unit-test-pull-secret" ],
405
- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
405
+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
406
406
write_to_file = True ,
407
407
appwrapper = False ,
408
408
local_queue = "local-queue-default" ,
@@ -430,7 +430,7 @@ def test_default_cluster_creation(mocker):
430
430
)
431
431
default_config = ClusterConfiguration (
432
432
name = "unit-test-default-cluster" ,
433
- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
433
+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
434
434
appwrapper = True ,
435
435
)
436
436
cluster = Cluster (default_config )
@@ -779,7 +779,7 @@ def test_ray_job_wrapping(mocker):
779
779
return_value = get_local_queue ("kueue.x-k8s.io" , "v1beta1" , "ns" , "localqueues" ),
780
780
)
781
781
cluster = cluster = createClusterWithConfig (mocker )
782
- cluster .config .image = "quay.io/project-codeflare/ray:latest -py39-cu118"
782
+ cluster .config .image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118"
783
783
mocker .patch (
784
784
"ray.job_submission.JobSubmissionClient._check_connection_and_version_with_url" ,
785
785
return_value = "None" ,
@@ -899,7 +899,7 @@ def test_ray_details(mocker, capsys):
899
899
ClusterConfiguration (
900
900
name = "raytest2" ,
901
901
namespace = "ns" ,
902
- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
902
+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
903
903
write_to_file = True ,
904
904
appwrapper = True ,
905
905
local_queue = "local_default_queue" ,
@@ -2292,7 +2292,7 @@ def test_cluster_status(mocker):
2292
2292
ClusterConfiguration (
2293
2293
name = "test" ,
2294
2294
namespace = "ns" ,
2295
- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
2295
+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
2296
2296
write_to_file = True ,
2297
2297
appwrapper = True ,
2298
2298
local_queue = "local_default_queue" ,
@@ -2387,7 +2387,7 @@ def test_wait_ready(mocker, capsys):
2387
2387
ClusterConfiguration (
2388
2388
name = "test" ,
2389
2389
namespace = "ns" ,
2390
- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
2390
+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
2391
2391
write_to_file = True ,
2392
2392
appwrapper = True ,
2393
2393
local_queue = "local-queue-default" ,
@@ -2614,7 +2614,7 @@ def throw_if_getting_raycluster(group, version, namespace, plural):
2614
2614
cluster = Cluster (
2615
2615
ClusterConfiguration (
2616
2616
"test_cluster" ,
2617
- image = "quay.io/project-codeflare/ray:latest -py39-cu118" ,
2617
+ image = "quay.io/project-codeflare/ray:2.20.0 -py39-cu118" ,
2618
2618
write_to_file = False ,
2619
2619
)
2620
2620
)
0 commit comments