From 761795c5c3c65fbae1f2a9a9ccd98f65a49d242c Mon Sep 17 00:00:00 2001
From: Saransh Chopra <saransh0701@gmail.com>
Date: Wed, 20 Nov 2024 13:32:00 +0000
Subject: [PATCH] fix axis

---
 glass/core/algorithm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/glass/core/algorithm.py b/glass/core/algorithm.py
index f54fbfbf..c13477df 100644
--- 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