Fix test for available resources (#2914)

This commit is contained in:
Peter Schafhalter 2018-09-25 23:07:23 -07:00 committed by Robert Nishihara
parent 971df5ea8a
commit fcdca6de18

View file

@ -26,7 +26,7 @@ class TestAvailableResources(object):
def test_no_tasks(self):
cluster_resources = ray.global_state.cluster_resources()
available_resources = ray.global_state.cluster_resources()
available_resources = ray.global_state.available_resources()
assert cluster_resources == available_resources
def test_replenish_resources(self):
@ -37,8 +37,8 @@ class TestAvailableResources(object):
resources_reset = False
while not resources_reset and time.time() - start < self.timeout:
resources_reset = (
cluster_resources == ray.global_state.available_resources())
available_resources = ray.global_state.available_resources()
resources_reset = (cluster_resources == available_resources)
assert resources_reset