From f936ea35fecb721d9cdb9ef21b662e4d8b91fb07 Mon Sep 17 00:00:00 2001 From: Ian Rodney Date: Fri, 13 Nov 2020 00:42:16 -0800 Subject: [PATCH] [hotfix] Fix ResourceDemandScheduler (#11996) * [hotfix] Fix ResourceDemandScheduler * fix test_autoscaler --- python/ray/tests/test_autoscaler.py | 7 ++++--- python/ray/tests/test_resource_demand_scheduler.py | 1 + 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/python/ray/tests/test_autoscaler.py b/python/ray/tests/test_autoscaler.py index ee17a3455..0e3478cde 100644 --- a/python/ray/tests/test_autoscaler.py +++ b/python/ray/tests/test_autoscaler.py @@ -1526,9 +1526,10 @@ MemTotal: 16396056 kB MemFree: 12869528 kB MemAvailable: 33000000 kB """ - runner.respond_to_call("cat /proc/meminfo", [proc_meminfo]) - runner.respond_to_call(".Runtimes", ["nvidia-container-runtime"]) - runner.respond_to_call("nvidia-smi", ["works"]) + runner.respond_to_call("cat /proc/meminfo", 2 * [proc_meminfo]) + runner.respond_to_call(".Runtimes", 2 * ["nvidia-container-runtime"]) + runner.respond_to_call("nvidia-smi", 2 * ["works"]) + runner.respond_to_call("json .Config.Env", 2 * ["[]"]) lm = LoadMetrics() autoscaler = StandardAutoscaler( config_path, diff --git a/python/ray/tests/test_resource_demand_scheduler.py b/python/ray/tests/test_resource_demand_scheduler.py index db9fa6b5b..2b7c61042 100644 --- a/python/ray/tests/test_resource_demand_scheduler.py +++ b/python/ray/tests/test_resource_demand_scheduler.py @@ -1033,6 +1033,7 @@ class AutoscalingTest(unittest.TestCase): config_path = self.write_config(config) self.provider = MockProvider() runner = MockProcessRunner() + runner.respond_to_call("json .Config.Env", ["[]"]) get_or_create_head_node( config, printable_config_file=config_path,