[Autoscaler] Fix resource passing bug fix (#10397)

This commit is contained in:
Alex Wu
2020-08-28 15:43:18 -07:00
committed by GitHub
parent 2afb54c99c
commit b1f3c9e10e
5 changed files with 32 additions and 26 deletions
@@ -366,10 +366,10 @@ class AutoscalingTest(unittest.TestCase):
# These checks are done separately because we have no guarantees on the
# order the dict is serialized in.
runner.assert_has_call("172.0.0.0", "RAY_OVERRIDE_RESOURCES=")
runner.assert_has_call("172.0.0.0", "CPU: 2")
runner.assert_has_call("172.0.0.0", "\"CPU\":2")
runner.assert_has_call("172.0.0.1", "RAY_OVERRIDE_RESOURCES=")
runner.assert_has_call("172.0.0.1", "CPU: 32")
runner.assert_has_call("172.0.0.1", "GPU: 8")
runner.assert_has_call("172.0.0.1", "\"CPU\":32")
runner.assert_has_call("172.0.0.1", "\"GPU\":8")
def testScaleUpLoadMetrics(self):
config = MULTI_WORKER_CLUSTER.copy()