From 25c9f25d79ea3962f1cb7c0fcbd227887e87695c Mon Sep 17 00:00:00 2001 From: Tony Kao Date: Thu, 12 Dec 2024 05:33:45 -0800 Subject: [PATCH] torchx - fix RaySchedulerTest unit test failure Summary: fix unit test failure on ray_scheduler_test due to missing required cfg arg on _validate() Differential Revision: D67140440 --- torchx/schedulers/test/ray_scheduler_test.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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,