mirror of
https://github.com/wassname/ray.git
synced 2026-07-09 04:00:44 +08:00
Fix CI test failures (#4007)
This commit is contained in:
+3
-3
@@ -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]
|
||||
|
||||
Reference in New Issue
Block a user