wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 05:43:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Deflake object manager test (#11052)

Browse Source
This commit is contained in:
Eric Liang
2020-09-26 18:33:02 -07:00
committed by GitHub
parent 51038f2197
commit b5ecdd088b
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tests/test_object_manager.py
+9 -2
View File
@@ -209,7 +209,8 @@ def test_object_transfer_retry(ray_start_cluster):
config = {
"object_manager_repeated_push_delay_ms": repeated_push_delay * 1000,
"object_manager_pull_timeout_ms": repeated_push_delay * 1000 / 4,
"object_manager_default_chunk_size": 1000
"object_manager_default_chunk_size": 1000,
"object_store_full_max_retries": 1,
}
object_store_memory = 150 * 1024 * 1024
cluster.add_node(
@@ -242,8 +243,14 @@ def test_object_transfer_retry(ray_start_cluster):
return not ray.worker.global_worker.core_worker.object_exists(x_id)
def force_eviction():
refs = []
for _ in range(20):
ray.put(np.zeros(object_store_memory // 10, dtype=np.uint8))
try:
refs.append(
ray.put(
np.zeros(object_store_memory // 10, dtype=np.uint8)))
except Exception:
break
wait_for_condition(not_exists)
# Force the object to be evicted from the local node.