diff --git a/torchx/schedulers/test/ray_scheduler_test.py b/torchx/schedulers/test/ray_scheduler_test.py index f8bde7e7c..92482d817 100644 --- 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,