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()