[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

engine/clients/chroma/configure.py

@@ -35,7 +35,10 @@ def clean(self):
 
     def recreate(self, dataset: Dataset, collection_params):
         params = self.collection_params
-        params["metadata"] = dict({"hnsw:space": self.DISTANCE_MAPPING.get(dataset.config.distance)}, **params.get('metadata', {}))
+        params["metadata"] = dict(
+            {"hnsw:space": self.DISTANCE_MAPPING.get(dataset.config.distance)},
+            **params.get("metadata", {}),
+        )
         self.client.create_collection(
             name=CHROMA_COLLECTION_NAME,
             **params,

engine/clients/chroma/parser.py

@@ -9,7 +9,9 @@
 
 class ChromaConditionParser(BaseConditionParser):
     def build_condition(
-        self, and_subfilters: Optional[List[Where]], or_subfilters: Optional[List[Where]]
+        self,
+        and_subfilters: Optional[List[Where]],
+        or_subfilters: Optional[List[Where]],
     ) -> Where:
         condition: Where = {}
         if and_subfilters is not None:
@@ -44,7 +46,9 @@ def build_range_filter(
             "$lte": lte,
             "$gte": gte,
         }
-        filters: OperatorExpression = {k: v for k, v in raw_filters.items() if v is not None}
+        filters: OperatorExpression = {
+            k: v for k, v in raw_filters.items() if v is not None
+        }
         return {field_name: filters}
 
     def build_geo_filter(

engine/clients/chroma/search.py

@@ -1,6 +1,6 @@
 from typing import List, Tuple
 
-from chromadb import HttpClient, ClientAPI, Settings
+from chromadb import ClientAPI, HttpClient, Settings
 from chromadb.api.types import IncludeEnum
 
 from dataset_reader.base_reader import Query
@@ -32,10 +32,13 @@ def search_one(cls, query: Query, top: int) -> List[Tuple[int, float]]:
             include=[IncludeEnum.distances],
         )
 
-        return [(int(hit[0]), float(hit[1])) for hit in zip(res["ids"][0], res["distances"][0])]
+        return [
+            (int(hit[0]), float(hit[1]))
+            for hit in zip(res["ids"][0], res["distances"][0])
+        ]
 
     def setup_search(self):
         metadata = self.collection.metadata.copy()
-        metadata.pop("hnsw:space", None) # Not allowed in the collection.modify method
-        metadata.update(self.search_params.get('metadata', {}))
+        metadata.pop("hnsw:space", None)  # Not allowed in the collection.modify method
+        metadata.update(self.search_params.get("metadata", {}))
         self.collection.modify(metadata=metadata)

engine/clients/chroma/upload.py

@@ -1,6 +1,6 @@
 from typing import List
 
-from chromadb import HttpClient, ClientAPI, Settings
+from chromadb import ClientAPI, HttpClient, Settings
 
 from dataset_reader.base_reader import Record
 from engine.base_client.upload import BaseUploader

experiments/configurations/chroma-single-node.md

@@ -23,5 +23,5 @@ See https://cookbook.chromadb.dev/core/configuration/#hnsw-configuration
 non-default not in use.
 
     "parallel": 16                      # implemented in base_client
-    "batch_size": 1024                  # implemented in base_client 
+    "batch_size": 1024                  # implemented in base_client