From a54d3a1cee8a655e269df823ecf679fb4c8d1f22 Mon Sep 17 00:00:00 2001 From: Alex Wu Date: Fri, 7 Aug 2020 13:30:39 -0700 Subject: [PATCH] [Autoscaler] Run resource demand scheduler in CI (#9945) * . * . --- ci/travis/ci.sh | 1 + python/ray/tests/BUILD | 1 + python/ray/tests/test_resource_demand_scheduler.py | 3 ++- 3 files changed, 4 insertions(+), 1 deletion(-) diff --git a/ci/travis/ci.sh b/ci/travis/ci.sh index e18d96bc1..4c78ee1be 100755 --- a/ci/travis/ci.sh +++ b/ci/travis/ci.sh @@ -155,6 +155,7 @@ test_python() { -python/ray/tests:test_object_manager -python/ray/tests:test_projects -python/ray/tests:test_ray_init # test_redis_port() seems to fail here, but pass in isolation + -python/ray/tests:test_resource_demand_scheduler -python/ray/tests:test_stress # timeout -python/ray/tests:test_stress_sharded # timeout -python/ray/tests:test_webui diff --git a/python/ray/tests/BUILD b/python/ray/tests/BUILD index e8ac4c6cf..2d9c569aa 100644 --- a/python/ray/tests/BUILD +++ b/python/ray/tests/BUILD @@ -41,6 +41,7 @@ py_test_module_list( "test_output.py", "test_placement_group.py", "test_reconstruction.py", + "test_resource_demand_scheduler.py", "test_reference_counting_2.py", "test_reference_counting.py", "test_stress.py", diff --git a/python/ray/tests/test_resource_demand_scheduler.py b/python/ray/tests/test_resource_demand_scheduler.py index 18f0f1f02..58920e28b 100644 --- a/python/ray/tests/test_resource_demand_scheduler.py +++ b/python/ray/tests/test_resource_demand_scheduler.py @@ -146,11 +146,12 @@ def test_get_instances_respects_max_limit(): class AutoscalingTest(unittest.TestCase): def setUp(self): NODE_PROVIDERS["mock"] = \ - lambda: (None, self.create_provider) + lambda config: self.create_provider self.provider = None self.tmpdir = tempfile.mkdtemp() def tearDown(self): + self.provider = None del NODE_PROVIDERS["mock"] shutil.rmtree(self.tmpdir) ray.shutdown()