From c2cbb85a43776b9624119476dec04f4823505445 Mon Sep 17 00:00:00 2001 From: Robert Nishihara Date: Sun, 19 Jan 2020 20:59:03 -0800 Subject: [PATCH] Fix flaky test test_feature_flag (#6850) --- python/ray/tests/test_reference_counting.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/ray/tests/test_reference_counting.py b/python/ray/tests/test_reference_counting.py index 84dd4336f..f1f9e2d76 100644 --- a/python/ray/tests/test_reference_counting.py +++ b/python/ray/tests/test_reference_counting.py @@ -231,10 +231,14 @@ def test_feature_flag(shutdown_only): self.large_object = ray.put( np.zeros(25 * 1024 * 1024, dtype=np.uint8)) + def wait_for_actor_to_start(self): + pass + def get_large_object(self): return ray.get(self.large_object) actor = Actor.remote() + ray.get(actor.wait_for_actor_to_start.remote()) for batch in range(10): intermediate_result = f.remote(