hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix test_dying_worker_get (#5908)

Browse source
This commit is contained in:
Eric Liang 2019-10-13 18:06:28 -07:00 • committed by Simon Mo
parent 0f24509c30
commit 2cbc67f3d5
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
python/ray/tests/test_component_failures.py
View file

@ -26,6 +26,7 @@ from ray.tests.utils import (run_string_as_driver_nonblocking,
def test_dying_worker_get(ray_start_2_cpus):
@ray.remote
def sleep_forever():
ray.experimental.signal.send("ready")
time.sleep(10**6)
@ray.remote
@ -33,7 +34,7 @@ def test_dying_worker_get(ray_start_2_cpus):
return os.getpid()
x_id = sleep_forever.remote()
time.sleep(0.01) # Try to wait for the sleep task to get scheduled.
ray.experimental.signal.receive([x_id]) # Block until it is scheduled.
# Get the PID of the other worker.
worker_pid = ray.get(get_worker_pid.remote())