Fix test

chromadb/test/property/test_filtering.py

@@ -13,6 +13,7 @@
     Metadatas,
     Where,
     WhereDocument,
+    Omitted,
 )
 import chromadb.test.property.strategies as strategies
 import hypothesis.strategies as st
@@ -345,7 +346,7 @@ def test_empty_filter(api: ServerAPI) -> None:
         n_results=3,
     )
     assert res["ids"] == [[], []]
-    assert res["embeddings"] is None
+    assert isinstance(res["embeddings"], Omitted)
     assert res["distances"] == [[], []]
     assert res["metadatas"] == [[], []]
 

chromadb/test/test_api.py

@@ -5,7 +5,7 @@
 
 import chromadb
 from chromadb.api.fastapi import FastAPI
-from chromadb.api.types import QueryResult, EmbeddingFunction, Document
+from chromadb.api.types import QueryResult, EmbeddingFunction, Document, Omitted
 from chromadb.config import Settings
 import chromadb.server.fastapi
 import pytest
@@ -92,7 +92,7 @@ def test_persist_index_loading(api_fixture, request):
         if (key in includes) or (key == "ids"):
             assert len(nn[key]) == 1
         else:
-            assert nn[key] is None
+            assert isinstance(nn[key], Omitted)
 
 
 @pytest.mark.parametrize("api_fixture", [local_persist_api])
@@ -119,7 +119,7 @@ def __call__(self, input):
         if (key in includes) or (key == "ids"):
             assert len(nn[key]) == 1
         else:
-            assert nn[key] is None
+            assert isinstance(nn[key], Omitted)
 
 
 @pytest.mark.parametrize("api_fixture", [local_persist_api])
@@ -1173,7 +1173,7 @@ def test_persist_index_loading_params(api, request):
         if (key in includes) or (key == "ids"):
             assert len(nn[key]) == 1
         else:
-            assert nn[key] is None
+            assert isinstance(nn[key], Omitted)
 
 
 def test_add_large(api):