fix: test

pawelpaszki · pawelpaszki · commit 874ede46913f · 2025-08-26T11:25:06.000+01:00
diff --git a/tests/e2e/rayjob_existing_cluster_kind_test.py b/tests/e2e/rayjob_existing_cluster_kind_test.py
@@ -95,7 +95,8 @@ def assert_rayjob_submit_against_existing_cluster(
                 "pip": "./tests/e2e/mnist_pip_requirements.txt",
                 "env_vars": get_setup_env_variables(ACCELERATOR=accelerator),
             },
-            shutdown_after_job_finishes=False,  # Don't shutdown the existing cluster
+            shutdown_after_job_finishes=False,
+            entrypoint_num_gpus=number_of_gpus,
         )
 
         # Submit the job

Original file line number	Diff line number	Diff line change
`@@ -95,7 +95,8 @@ def assert_rayjob_submit_against_existing_cluster(`
`95`	`95`	`"pip": "./tests/e2e/mnist_pip_requirements.txt",`
`96`	`96`	`"env_vars": get_setup_env_variables(ACCELERATOR=accelerator),`
`97`	`97`	`},`
`98`		`- shutdown_after_job_finishes=False, # Don't shutdown the existing cluster`
	`98`	`+ shutdown_after_job_finishes=False,`
	`99`	`+ entrypoint_num_gpus=number_of_gpus,`
`99`	`100`	`)`
`100`	`101`
`101`	`102`	`# Submit the job`