black

qdrant · Mar 17, 2024 · c22f3fc · c22f3fc
1 parent e1f337d
commit c22f3fc
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 11 deletions.
diff --git a/engine/base_client/client.py b/engine/base_client/client.py
@@ -15,7 +15,7 @@
 
 DETAILED_RESULTS = bool(int(os.getenv("DETAILED_RESULTS", False)))
 
- 
+
 class BaseClient:
     def __init__(
         self,
@@ -42,15 +42,18 @@ def save_search_results(
         result_path = RESULTS_DIR / experiments_file
         with open(result_path, "w") as out:
             out.write(
-                json.dumps({
-                    "params": {
-                        "dataset": dataset_name,
-                        "experiment": self.name,
-                        "engine": self.engine,
-                        **search_params
+                json.dumps(
+                    {
+                        "params": {
+                            "dataset": dataset_name,
+                            "experiment": self.name,
+                            "engine": self.engine,
+                            **search_params,
+                        },
+                        "results": results,
                     },
-                    "results": results
-                }, indent=2)
+                    indent=2,
+                )
             )
         return result_path
 
@@ -66,7 +69,7 @@ def save_upload_results(
                     "experiment": self.name,
                     "engine": self.engine,
                     "dataset": dataset_name,
-                    **upload_params
+                    **upload_params,
                 },
                 "results": results,
             }

diff --git a/engine/clients/milvus/config.py b/engine/clients/milvus/config.py
@@ -10,7 +10,7 @@
     Distance.L2: "L2",
     Distance.DOT: "IP",
     # Milvus does not support cosine. Cosine is equal to IP of normalized vectors
-    Distance.COSINE: "IP"
+    Distance.COSINE: "IP",
     # Jaccard, Tanimoto, Hamming distance, Superstructure and Substructure are also available
 }