Fix prop test

chroma-core · Dec 11, 2024 · 90b3cb9 · 90b3cb9
1 parent 4452961
commit 90b3cb9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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]