Fix: Upcast to np.float32 on gets, if bf16 is used

unum-cloud · Aug 19, 2024 · 25b90f1 · 25b90f1
1 parent ca2111a
commit 25b90f1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/python/usearch/eval.py b/python/usearch/eval.py
@@ -124,7 +124,7 @@ def self_recall(index: Index, sample: Union[float, int] = 1.0, **kwargs) -> Sear
     if "vectors" in kwargs:
         vectors = kwargs.pop("vectors")
     else:
-        vectors = index.get(keys, index.dtype)
+        vectors = index.get(keys)
 
     matches = index.search(vectors, **kwargs)
     count_matches: int = (

diff --git a/python/usearch/index.py b/python/usearch/index.py
@@ -128,6 +128,8 @@ def _normalize_dtype(
 
 
 def _to_numpy_dtype(dtype: ScalarKind):
+    if dtype == ScalarKind.BF16:
+        return None
     _normalize = {
         ScalarKind.F64: np.float64,
         ScalarKind.F32: np.float32,
@@ -738,10 +740,15 @@ def get(
         """
         if not dtype:
             dtype = self.dtype
+            view_dtype = _to_numpy_dtype(dtype)
+            if view_dtype is None:
+                dtype = ScalarKind.F32
+                view_dtype = np.float32
         else:
             dtype = _normalize_dtype(dtype)
-
-        view_dtype = _to_numpy_dtype(dtype)
+            view_dtype = _to_numpy_dtype(dtype)
+            if view_dtype is None:
+                raise NotImplementedError("The requested representation type is not supported by NumPy")
 
         def cast(result):
             if result is not None: