Fix the grad weighted minkowski as well...

lmcinnes · Feb 9, 2022 · c9628e5 · c9628e5
1 parent 28bea09
commit c9628e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/umap/distances.py b/umap/distances.py
@@ -256,7 +256,7 @@ def weighted_minkowski_grad(x, y, w=_mock_ones, p=2):
     """
     result = 0.0
     for i in range(x.shape[0]):
-        result += (w[i] * np.abs(x[i] - y[i])) ** p
+        result += w[i] * (np.abs(x[i] - y[i])) ** p
 
     grad = np.empty(x.shape[0], dtype=np.float32)
     for i in range(x.shape[0]):