File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -31,8 +31,11 @@ def test_rayjob_ray_cluster_sdk_kind_nvidia_gpu(self):
3131 self .setup_method ()
3232 create_namespace (self )
3333 create_kueue_resources (self )
34+ # self.run_rayjob_against_existing_cluster_kind(
35+ # accelerator="gpu", number_of_gpus=1
36+ # )
3437 self .run_rayjob_against_existing_cluster_kind (
35- accelerator = "gpu " , number_of_gpus = 1
38+ accelerator = "cpu " , number_of_gpus = 0
3639 )
3740
3841 def run_rayjob_against_existing_cluster_kind (
@@ -50,14 +53,14 @@ def run_rayjob_against_existing_cluster_kind(
5053 worker_cpu_limits = 1 ,
5154 worker_memory_requests = 1 ,
5255 worker_memory_limits = 4 ,
53- worker_extended_resource_requests = {gpu_resource_name : number_of_gpus },
56+ # worker_extended_resource_requests={gpu_resource_name: number_of_gpus},
5457 image = "rayproject/ray:2.47.1" ,
5558 write_to_file = True ,
5659 verify_tls = False ,
5760 )
5861 )
5962
60- cluster .apply ()
63+ cluster .up ()
6164 cluster .status ()
6265 cluster .wait_ready ()
6366 cluster .status ()
You can’t perform that action at this time.
0 commit comments