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(