fixed

KlugerLab · Aug 5, 2024 · b789260 · b789260
1 parent f810b70
commit b789260
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/dyson_equalizer/algorithm.py b/dyson_equalizer/algorithm.py
@@ -159,10 +159,10 @@ def scale_matrix(
         The normalized matrix
     """
     m, n = Y.shape[-2:]
-    if x.shape != [m]:
-        raise ValueError(f"The row weights vector have the wrong shape {x.shape}")
-    if y.shape != [n]:
-        raise ValueError(f"The columns weights vector have the wrong shape {y.shape}")
+    if x.shape != (m, ):
+        raise ValueError(f"The row weights vector have the wrong shape {x.shape}, expected {(m,)}")
+    if y.shape != (n, ):
+        raise ValueError(f"The columns weights vector have the wrong shape {y.shape}, expected {(n,)}")
 
     return x[:, None] * Y * y