fix axis

glass-dev · Nov 26, 2024 · 761795c · 761795c
1 parent 1f11e40
commit 761795c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/glass/core/algorithm.py b/glass/core/algorithm.py
@@ -83,7 +83,7 @@ def nnls(
     for _ in range(maxiter):
         if xp.all(p):
             break
-        w = xp.linalg.vecdot(b - a @ x, a, axis=1)
+        w = xp.linalg.vecdot(b - a @ x, a, axis=0)
         m = index[~p][xp.argmax(w[~p])]
         if w[m] <= tol:
             break