fix(metrics): cast to float to compute luv_approx

loiccoyle · Jun 24, 2024 · 8ef1e80 · 8ef1e80
1 parent 65d27dd
commit 8ef1e80
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/phomo/metrics.py b/phomo/metrics.py
@@ -6,8 +6,9 @@
 
 
 class MetricCallable(Protocol):  # type: ignore
-    def __call__(self, master_chunk: np.ndarray, tile_arrays: np.ndarray) -> np.ndarray:
-        ...
+    def __call__(
+        self, master_chunk: np.ndarray, tile_arrays: np.ndarray
+    ) -> np.ndarray: ...
 
 
 def register_metric(func):
@@ -107,6 +108,7 @@ def luv_approx(
             arrays.
     """
     r = (master_chunk[:, :, 0] + tile_arrays[:, :, :, 0]) // 2
+    r = r.astype(float)
     d = np.subtract(master_chunk, tile_arrays, dtype=float)
     return np.linalg.norm(
         (