Skip to content

Commit c11b7ad

Browse files
committed
[BUG] Replace empty config objects with None during deserializing
1 parent ec1f94c commit c11b7ad

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

chromadb/types.py

+7-9
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,6 @@
1616
from chromadb.serde import BaseModelJSONSerializable
1717
from chromadb.api.collection_configuration import (
1818
CollectionConfiguration,
19-
HNSWConfiguration,
20-
SpannConfiguration,
2119
collection_configuration_to_json,
2220
load_collection_configuration_from_json,
2321
)
@@ -155,8 +153,8 @@ def get_configuration(self) -> CollectionConfiguration:
155153
stacklevel=2,
156154
)
157155
return CollectionConfiguration(
158-
hnsw=HNSWConfiguration(),
159-
spann=SpannConfiguration(),
156+
hnsw=None,
157+
spann=None,
160158
embedding_function=None,
161159
)
162160

@@ -175,11 +173,11 @@ def get_model_fields(self) -> Dict[Any, Any]:
175173
@override
176174
def from_json(cls, json_map: Dict[str, Any]) -> Self:
177175
"""Deserializes a Collection object from JSON"""
178-
configuration: CollectionConfiguration = {
179-
"hnsw": {},
180-
"spann": {},
181-
"embedding_function": None,
182-
}
176+
configuration = CollectionConfiguration(
177+
hnsw=None,
178+
spann=None,
179+
embedding_function=None,
180+
)
183181
try:
184182
configuration_json = json_map.get("configuration_json", None)
185183
configuration = load_collection_configuration_from_json(configuration_json)

0 commit comments

Comments
 (0)