Fix flaky test test_feature_flag (#6850)

2026-06-29 06:33:06 +08:00 · 2020-01-19 20:59:03 -08:00
parent 341ddd0a09
commit c2cbb85a43
1 changed files with 4 additions and 0 deletions
@@ -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(