[Autoscaler] Run resource demand scheduler in CI (#9945)

* . * .
2026-07-21 12:50:45 +08:00 · 2020-08-07 13:30:39 -07:00
parent ca37accc07
commit a54d3a1cee
3 changed files with 4 additions and 1 deletions
@@ -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()