mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 18:11:38 +08:00
Fix queue actor init in setup_queue_actor fixture (#5676)
This commit is contained in:
committed by
Robert Nishihara
parent
3ed18d0b59
commit
fc9f03cd96
@@ -1776,7 +1776,11 @@ def setup_queue_actor():
|
||||
def read(self):
|
||||
return self.queue
|
||||
|
||||
yield Queue.remote()
|
||||
queue = Queue.remote()
|
||||
# Make sure queue actor is initialized.
|
||||
ray.get(queue.read.remote())
|
||||
|
||||
yield queue
|
||||
|
||||
# The code after the yield will run as teardown code.
|
||||
ray.shutdown()
|
||||
@@ -1784,8 +1788,6 @@ def setup_queue_actor():
|
||||
|
||||
def test_fork(setup_queue_actor):
|
||||
queue = setup_queue_actor
|
||||
# Make sure queue actor is initialized.
|
||||
ray.get(queue.read.remote())
|
||||
|
||||
@ray.remote
|
||||
def fork(queue, key, item):
|
||||
|
||||
Reference in New Issue
Block a user