Skip to content

Commit c1a195a

Browse files
authored
torchx - fix RaySchedulerTest unit test failure
Differential Revision: D67140440 Pull Request resolved: pytorch#990
1 parent 617b1ab commit c1a195a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

torchx/schedulers/test/ray_scheduler_test.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,9 @@ def test_validate_warns_when_role_contains_port_map(self) -> None:
197197
with self._assert_log_message(
198198
"WARNING", "The Ray scheduler does not support port mapping."
199199
):
200-
self._scheduler._validate(self._app_def, scheduler="ray")
200+
self._scheduler._validate(
201+
self._app_def, scheduler="ray", cfg=self._run_cfg
202+
)
201203

202204
def test_submit_dryrun_raises_error_if_cluster_config_file_is_not_str(
203205
self,

0 commit comments

Comments
 (0)