mirror of
https://github.com/wassname/ray.git
synced 2026-07-18 12:40:56 +08:00
Remove experimental.signal API (#7477)
* Remove experimental.signal API * fix test
This commit is contained in:
@@ -1,12 +1,12 @@
|
||||
import os
|
||||
import signal
|
||||
from signal import SIGKILL
|
||||
import sys
|
||||
import time
|
||||
|
||||
import pytest
|
||||
|
||||
import ray
|
||||
from ray.test_utils import run_string_as_driver_nonblocking
|
||||
from ray.test_utils import run_string_as_driver_nonblocking, SignalActor
|
||||
|
||||
|
||||
# This test checks that when a worker dies in the middle of a get, the plasma
|
||||
@@ -16,16 +16,18 @@ from ray.test_utils import run_string_as_driver_nonblocking
|
||||
reason="Not working with new GCS API.")
|
||||
def test_dying_worker_get(ray_start_2_cpus):
|
||||
@ray.remote
|
||||
def sleep_forever():
|
||||
ray.experimental.signal.send("ready")
|
||||
def sleep_forever(signal):
|
||||
ray.get(signal.send.remote())
|
||||
time.sleep(10**6)
|
||||
|
||||
@ray.remote
|
||||
def get_worker_pid():
|
||||
return os.getpid()
|
||||
|
||||
x_id = sleep_forever.remote()
|
||||
ray.experimental.signal.receive([x_id]) # Block until it is scheduled.
|
||||
signal = SignalActor.remote()
|
||||
|
||||
x_id = sleep_forever.remote(signal)
|
||||
ray.get(signal.wait.remote())
|
||||
# Get the PID of the other worker.
|
||||
worker_pid = ray.get(get_worker_pid.remote())
|
||||
|
||||
@@ -42,7 +44,7 @@ def test_dying_worker_get(ray_start_2_cpus):
|
||||
assert len(ready_ids) == 0
|
||||
|
||||
# Kill the worker.
|
||||
os.kill(worker_pid, signal.SIGKILL)
|
||||
os.kill(worker_pid, SIGKILL)
|
||||
time.sleep(0.1)
|
||||
|
||||
# Make sure the sleep task hasn't finished.
|
||||
@@ -129,7 +131,7 @@ def test_dying_worker_wait(ray_start_2_cpus):
|
||||
time.sleep(0.1)
|
||||
|
||||
# Kill the worker.
|
||||
os.kill(worker_pid, signal.SIGKILL)
|
||||
os.kill(worker_pid, SIGKILL)
|
||||
time.sleep(0.1)
|
||||
|
||||
# Create the object.
|
||||
|
||||
Reference in New Issue
Block a user