From f0fe37ed7f122c8a2fbf249b1e1596837a1eff2b Mon Sep 17 00:00:00 2001 From: SangBin Cho Date: Mon, 30 Oct 2023 15:34:38 +0900 Subject: [PATCH] [Core] Fix flaky reference counting #40783 Co-authored-by: SangBin Cho --- python/ray/tests/test_reference_counting.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/ray/tests/test_reference_counting.py b/python/ray/tests/test_reference_counting.py index 72e8502fa171..c16097453d5b 100644 --- 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, )