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,