From fc9f03cd96908bd2ab21847991807f5352e820f6 Mon Sep 17 00:00:00 2001 From: Simon Mo Date: Fri, 13 Sep 2019 12:35:44 -0700 Subject: [PATCH] Fix queue actor init in setup_queue_actor fixture (#5676) --- python/ray/tests/test_actor.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/python/ray/tests/test_actor.py b/python/ray/tests/test_actor.py index 7189ca0fa..e994210e1 100644 --- a/python/ray/tests/test_actor.py +++ b/python/ray/tests/test_actor.py @@ -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):