usearch 0.18.8 → 0.19.1

package/python/scripts/bench.py

@@ -24,15 +24,13 @@ def bench_speed(
     jit: bool = False,
     train: bool = False,
 ) -> pd.DataFrame:
-
     # Build various indexes:
     indexes = []
     jit_options = [False]
     if jit:
         jit_options.append(jit)
-    dtype_options = ['f32', 'f16', 'f8']
+    dtype_options = ["f32", "f16", "f8"]
     for jit, dtype in itertools.product(jit_options, dtype_options):
-
         index = Index(
             ndim=eval.ndim,
             jit=jit,
@@ -40,9 +38,9 @@ def bench_speed(
             expansion_add=expansion_add,
             expansion_search=expansion_search,
             connectivity=connectivity,
-            path='USearch' + ['', '+JIT'][jit] + ':' + dtype,
+            path="USearch" + ["", "+JIT"][jit] + ":" + dtype,
         )
-        
+
         # Skip the cases, where JIT-ing is impossible
         if jit and not index.jit:
             continue
@@ -51,41 +49,60 @@ def bench_speed(
     # Add FAISS indexes to the mix:
     try:
         from index_faiss import IndexFAISS, IndexQuantizedFAISS
-        indexes.append(IndexFAISS(
-            ndim=eval.ndim,
-            expansion_add=expansion_add,
-            expansion_search=expansion_search,
-            connectivity=connectivity,
-            path='FAISS:f32',
-        ))
-        if train:
-            indexes.append(IndexQuantizedFAISS(
-                train=eval.tasks[0].vectors,
+
+        indexes.append(
+            IndexFAISS(
+                ndim=eval.ndim,
                 expansion_add=expansion_add,
                 expansion_search=expansion_search,
                 connectivity=connectivity,
-                path='FAISS+IVFPQ:f32',
-            ))
+                path="FAISS:f32",
+            )
+        )
+        if train:
+            indexes.append(
+                IndexQuantizedFAISS(
+                    train=eval.tasks[0].vectors,
+                    expansion_add=expansion_add,
+                    expansion_search=expansion_search,
+                    connectivity=connectivity,
+                    path="FAISS+IVFPQ:f32",
+                )
+            )
     except (ImportError, ModuleNotFoundError):
         pass
 
     # Time to evaluate:
     results = [eval(index) for index in indexes]
-    return pd.DataFrame({
-        'names': [i.path for i in indexes],
-        'add_per_second': [x['add_per_second'] for x in results],
-        'search_per_second': [x['search_per_second'] for x in results],
-        'recall_at_one': [x['recall_at_one'] for x in results],
-    })
+    return pd.DataFrame(
+        {
+            "names": [i.path for i in indexes],
+            "add_per_second": [x["add_per_second"] for x in results],
+            "search_per_second": [x["search_per_second"] for x in results],
+            "recall_at_one": [x["recall_at_one"] for x in results],
+        }
+    )
 
 
 def bench_params(
     count: int = 1_000_000,
     connectivities: int = range(10, 20),
     dimensions: List[int] = [
-        2, 3, 4, 8, 16, 32, 96, 100,
-        256, 384, 512, 768, 1024, 1536],
-
+        2,
+        3,
+        4,
+        8,
+        16,
+        32,
+        96,
+        100,
+        256,
+        384,
+        512,
+        768,
+        1024,
+        1536,
+    ],
     expansion_add: int = DEFAULT_EXPANSION_ADD,
     expansion_search: int = DEFAULT_EXPANSION_SEARCH,
 ) -> pd.DataFrame:
@@ -97,7 +114,6 @@ def bench_params(
 
     results = []
     for connectivity, ndim in itertools.product(connectivities, dimensions):
-
         task = AddTask(
             labels=np.arange(count, dtype=Label),
             vectors=np.random.rand(count, ndim).astype(np.float32),
@@ -109,8 +125,8 @@ def bench_params(
             expansion_search=expansion_search,
         )
         result = asdict(task(index))
-        result['ndim'] = dimensions
-        result['connectivity'] = connectivity
+        result["ndim"] = dimensions
+        result["connectivity"] = connectivity
         results.append(result)
 
     # return self._execute_tasks(

package/python/scripts/index_faiss.py

@@ -13,7 +13,6 @@ from usearch.index import (
 
 
 class IndexFAISS:
-
     def __init__(
         self,
         index: IndexHNSWFlat = None,
@@ -22,9 +21,9 @@ class IndexFAISS:
         expansion_add: int = DEFAULT_EXPANSION_ADD,
         expansion_search: int = DEFAULT_EXPANSION_SEARCH,
         path: Optional[os.PathLike] = None,
-        *args, **kwargs,
+        *args,
+        **kwargs,
     ):
-
         if index is None:
             index = IndexHNSWFlat(ndim, connectivity)
             index.hnsw.efConstruction = expansion_add
@@ -32,11 +31,11 @@ class IndexFAISS:
 
         self._faiss = index
         self._specs = {
-            'Class': 'usearch.IndexFAISS',
-            'Dimensions': ndim,
-            'Connectivity': connectivity,
-            'Expansion@Add': expansion_add,
-            'Expansion@Search': expansion_search,
+            "Class": "usearch.IndexFAISS",
+            "Dimensions": ndim,
+            "Connectivity": connectivity,
+            "Expansion@Add": expansion_add,
+            "Expansion@Search": expansion_search,
         }
 
         self.path = path
@@ -58,9 +57,11 @@ class IndexFAISS:
 
     @property
     def specs(self) -> dict:
-        self._specs.update({
-            'Size': len(self),
-        })
+        self._specs.update(
+            {
+                "Size": len(self),
+            }
+        )
         return self._specs
 
     def load(self, path: os.PathLike):
@@ -68,33 +69,32 @@ class IndexFAISS:
 
 
 class IndexQuantizedFAISS(IndexFAISS):
-
     def __init__(
         self,
         train: np.ndarray,
         connectivity: int = DEFAULT_CONNECTIVITY,
         expansion_add: int = DEFAULT_EXPANSION_ADD,
         expansion_search: int = DEFAULT_EXPANSION_SEARCH,
-        *args, **kwargs,
+        *args,
+        **kwargs,
     ):
-
         ndim = train.shape[1]
         super().__init__(
             ndim=ndim,
             connectivity=connectivity,
             expansion_add=expansion_add,
             expansion_search=expansion_search,
-            *args, **kwargs,
+            *args,
+            **kwargs,
         )
 
         nlist = 10000  # Number of inverted lists (number of partitions).
         nsegment = 16  # Number of segments for PQ (number of subquantizers).
-        nbit = 8       # Number of bits to encode each segment.
+        nbit = 8  # Number of bits to encode each segment.
 
         self._original_faiss = self._faiss
-        self._faiss = IndexIVFPQ(
-            self._faiss, ndim, nlist, nsegment, nbit)
+        self._faiss = IndexIVFPQ(self._faiss, ndim, nlist, nsegment, nbit)
 
         self._faiss.train(train)
         self._faiss.nprobe = 10
-        self._specs['Class'] = 'usearch.IndexQuantizedFAISS'
+        self._specs["Class"] = "usearch.IndexQuantizedFAISS"