Fix CI test failures (#4007)

This commit is contained in:
Yuhong Guo
2019-02-11 11:01:14 +08:00
committed by Hao Chen
parent e703b9f49d
commit 5fb1efd60d
4 changed files with 16 additions and 16 deletions
+3 -3
View File
@@ -1074,7 +1074,7 @@ def test_actors_and_tasks_with_gpus(ray_start_cluster):
@ray.remote(num_gpus=1)
def f1():
t1 = time.monotonic()
time.sleep(0.2)
time.sleep(0.4)
t2 = time.monotonic()
gpu_ids = ray.get_gpu_ids()
assert len(gpu_ids) == 1
@@ -1085,7 +1085,7 @@ def test_actors_and_tasks_with_gpus(ray_start_cluster):
@ray.remote(num_gpus=2)
def f2():
t1 = time.monotonic()
time.sleep(0.2)
time.sleep(0.4)
t2 = time.monotonic()
gpu_ids = ray.get_gpu_ids()
assert len(gpu_ids) == 2
@@ -1186,7 +1186,7 @@ def test_actors_and_tasks_with_gpus_version_two(shutdown_only):
@ray.remote(num_gpus=1)
def f():
time.sleep(4)
time.sleep(5)
gpu_ids = ray.get_gpu_ids()
assert len(gpu_ids) == 1
return gpu_ids[0]