torchx - fix RaySchedulerTest unit test failure

Differential Revision: D67140440 Pull Request resolved: #990
pytorch · Dec 12, 2024 · c1a195a · c1a195a
1 parent 617b1ab
commit c1a195a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torchx/schedulers/test/ray_scheduler_test.py b/torchx/schedulers/test/ray_scheduler_test.py
@@ -197,7 +197,9 @@ def test_validate_warns_when_role_contains_port_map(self) -> None:
             with self._assert_log_message(
                 "WARNING", "The Ray scheduler does not support port mapping."
             ):
-                self._scheduler._validate(self._app_def, scheduler="ray")
+                self._scheduler._validate(
+                    self._app_def, scheduler="ray", cfg=self._run_cfg
+                )
 
         def test_submit_dryrun_raises_error_if_cluster_config_file_is_not_str(
             self,