mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 15:40:09 +08:00
Remove timeouts in test_cancel (#8272)
This commit is contained in:
@@ -53,7 +53,7 @@ def test_cancel_chain(ray_start_regular, use_force):
|
||||
ray.get(obj2, timeout=.1)
|
||||
|
||||
signaler2.send.remote()
|
||||
ray.get(obj1, timeout=10)
|
||||
ray.get(obj1)
|
||||
|
||||
|
||||
@pytest.mark.parametrize("use_force", [True, False])
|
||||
@@ -89,7 +89,7 @@ def test_cancel_multiple_dependents(ray_start_regular, use_force):
|
||||
ray.get(d)
|
||||
|
||||
signaler.send.remote()
|
||||
ray.get(head2, timeout=1)
|
||||
ray.get(head2)
|
||||
|
||||
|
||||
@pytest.mark.parametrize("use_force", [True, False])
|
||||
@@ -109,7 +109,7 @@ def test_single_cpu_cancel(shutdown_only, use_force):
|
||||
assert len(ray.wait([obj3], timeout=.1)[0]) == 0
|
||||
ray.cancel(obj3, use_force)
|
||||
with pytest.raises(valid_exceptions(use_force)):
|
||||
ray.get(obj3, 10)
|
||||
ray.get(obj3)
|
||||
|
||||
ray.cancel(obj1, use_force)
|
||||
|
||||
@@ -151,7 +151,7 @@ def test_comprehensive(ray_start_regular, use_force):
|
||||
signaler.send.remote()
|
||||
|
||||
with pytest.raises(valid_exceptions(use_force)):
|
||||
ray.get(combo, 10)
|
||||
ray.get(combo)
|
||||
|
||||
|
||||
@pytest.mark.parametrize("use_force", [True, False])
|
||||
@@ -179,7 +179,7 @@ def test_stress(shutdown_only, use_force):
|
||||
|
||||
for done in cancelled:
|
||||
with pytest.raises(valid_exceptions(use_force)):
|
||||
ray.get(done, 10)
|
||||
ray.get(done)
|
||||
|
||||
for indx in range(len(tasks)):
|
||||
t = tasks[indx]
|
||||
@@ -188,7 +188,7 @@ def test_stress(shutdown_only, use_force):
|
||||
cancelled.add(t)
|
||||
if t in cancelled:
|
||||
with pytest.raises(valid_exceptions(use_force)):
|
||||
ray.get(t, 10)
|
||||
ray.get(t)
|
||||
else:
|
||||
ray.get(t)
|
||||
|
||||
@@ -223,7 +223,7 @@ def test_fast(shutdown_only, use_force):
|
||||
signaler.send.remote()
|
||||
for obj_id in ids:
|
||||
try:
|
||||
ray.get(obj_id, 10)
|
||||
ray.get(obj_id)
|
||||
except Exception as e:
|
||||
assert isinstance(e, valid_exceptions(use_force))
|
||||
|
||||
|
||||
Reference in New Issue
Block a user