From 06d33f6cc412fac3bf9d11457fca7f808f1a323c Mon Sep 17 00:00:00 2001 From: Kartik Date: Fri, 12 Jun 2026 21:07:47 +0530 Subject: [PATCH] fix: relax flaky entity boost parallelism timing threshold (#5511) --- tests/memory/test_main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/memory/test_main.py b/tests/memory/test_main.py index 677e7fe4c..6e3e75064 100644 --- a/tests/memory/test_main.py +++ b/tests/memory/test_main.py @@ -828,7 +828,7 @@ class TestEntityBoostParallelism: boosts = mock_memory._compute_entity_boosts(entities, {"user_id": "u1"}) elapsed = time.perf_counter() - start - assert elapsed < 0.6, f"searches did not run concurrently (took {elapsed:.2f}s)" + assert elapsed < 0.75, f"searches did not run concurrently (took {elapsed:.2f}s)" assert concurrent_count["peak"] >= 2, "no overlap observed between entity searches" assert len(boosts) == 4 @@ -854,6 +854,6 @@ class TestEntityBoostParallelism: boosts = await mock_async_memory._compute_entity_boosts_async(entities, {"user_id": "u1"}) elapsed = time.perf_counter() - start - assert elapsed < 0.6, f"searches did not run concurrently (took {elapsed:.2f}s)" + assert elapsed < 0.75, f"searches did not run concurrently (took {elapsed:.2f}s)" assert concurrent_count["peak"] >= 2, "no overlap observed between entity searches" assert len(boosts) == 4