[Core] Fix flaky reference counting ray-project#40783

Co-authored-by: SangBin Cho <[email protected]>
miqdigital · Oct 30, 2023 · f0fe37e · f0fe37e
1 parent 728766b
commit f0fe37e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/ray/tests/test_reference_counting.py b/python/ray/tests/test_reference_counting.py
@@ -45,7 +45,10 @@ def _fill_object_store_and_get(obj, succeed=True, object_MiB=20, num_objects=5):
         )
     else:
         wait_for_condition(
-            lambda: not ray._private.worker.global_worker.core_worker.object_exists(obj)
+            lambda: not ray._private.worker.global_worker.core_worker.object_exists(
+                obj
+            ),
+            timeout=30,
         )