From 0e2776c98a6637e1cbab5c1582214bfdbb861499 Mon Sep 17 00:00:00 2001 From: Ben Eggers Date: Wed, 13 Mar 2024 10:20:05 -0700 Subject: [PATCH] [BUG] Shut down test system to hopefully fix test flakiness --- chromadb/test/db/test_system.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/chromadb/test/db/test_system.py b/chromadb/test/db/test_system.py index 383b2949ab2..26fcc325497 100644 --- a/chromadb/test/db/test_system.py +++ b/chromadb/test/db/test_system.py @@ -120,6 +120,7 @@ def grpc_with_mock_server() -> Generator[SysDB, None, None]: system.start() client.reset_and_wait_for_ready() yield client + system.stop() def grpc_with_real_server() -> Generator[SysDB, None, None]: @@ -145,7 +146,8 @@ def db_fixtures() -> List[Callable[[], Generator[SysDB, None, None]]]: @pytest.fixture(scope="module", params=db_fixtures()) def sysdb(request: FixtureRequest) -> Generator[SysDB, None, None]: - yield next(request.param()) + sysdb = request.param() + yield next(sysdb) # region Collection tests