fix: Adapt to new list_collection semantics

chroma-core · Jan 3, 2025 · b53dadb · b53dadb
1 parent 00a7394
commit b53dadb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/chromadb/test/property/invariants.py b/chromadb/test/property/invariants.py
@@ -468,7 +468,7 @@ def log_size_for_collections_match_expected(
 @contextmanager
 def collection_deleted(client: ClientAPI, collection_name: str):
     # Invariant checks before deletion
-    assert collection_name in [c.name for c in client.list_collections()]
+    assert collection_name in client.list_collections()
     collection = client.get_collection(collection_name)
     segments = []
     if isinstance(client._server, SegmentAPI):  # type: ignore
@@ -501,7 +501,7 @@ def collection_deleted(client: ClientAPI, collection_name: str):
     yield
 
     # Invariant checks after deletion
-    assert collection_name not in [c.name for c in client.list_collections()]
+    assert collection_name not in client.list_collections()
     if len(segments) > 0:
         sysdb: SysDB = client._server._sysdb  # type: ignore
         segments_after = sysdb.get_segments(collection=collection.id)