Fix queue actor init in setup_queue_actor fixture (#5676)

This commit is contained in:
Simon Mo
2019-09-13 12:35:44 -07:00
committed by Robert Nishihara
parent 3ed18d0b59
commit fc9f03cd96
+5 -3
View File
@@ -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):