Fix flaky test test_feature_flag (#6850)

ray-project · Jan 20, 2020 · c2cbb85 · c2cbb85
1 parent 341ddd0
commit c2cbb85
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/ray/tests/test_reference_counting.py b/python/ray/tests/test_reference_counting.py
@@ -231,10 +231,14 @@ def __init__(self):
  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(