From 90b3cb97326d3a2ed32d168137699a278a034a69 Mon Sep 17 00:00:00 2001 From: Sicheng Pan Date: Wed, 11 Dec 2024 11:11:35 -0800 Subject: [PATCH] Fix prop test --- chromadb/test/property/test_sysdb.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/chromadb/test/property/test_sysdb.py b/chromadb/test/property/test_sysdb.py index db475cb28848..0f57b7949814 100644 --- a/chromadb/test/property/test_sysdb.py +++ b/chromadb/test/property/test_sysdb.py @@ -18,7 +18,7 @@ from chromadb.db.system import SysDB from chromadb.segment import SegmentType from chromadb.test.conftest import NOT_CLUSTER_ONLY -from chromadb.test.db.test_system import sysdb +from chromadb.test.db.test_system import sqlite, grpc_with_real_server from chromadb.types import Segment, SegmentScope @@ -153,5 +153,6 @@ def delete_collection(self, coll: strategies.Collection) -> None: self.sysdb.delete_collection(id=coll.id) -def test_sysdb(caplog: pytest.LogCaptureFixture, sysdb: SysDB, system: System) -> None: +def test_sysdb(caplog: pytest.LogCaptureFixture, system: System) -> None: + sysdb = next(sqlite()) if NOT_CLUSTER_ONLY else next(grpc_with_real_server()) run_state_machine_as_test(lambda: SysDBStateMachine(sysdb=sysdb)) # type: ignore[no-untyped-call]