make matvec compatible with scipy sparse arrays

FMatti · Jul 3, 2024 · eadee2b · eadee2b
1 parent 57923e0
commit eadee2b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/roughly/approximate/krylov.py b/roughly/approximate/krylov.py
@@ -23,7 +23,7 @@ def _preprocess(self, A, X, k, dtype=None):
         self.n, self.m = X.shape
         self.k = k
 
-        self.matvec = (lambda x: A @ x) if isinstance(A, np.ndarray) else A
+        self.matvec = (lambda x: A @ x) if isinstance(A, np.ndarray | sp.sparse.sparray) else A
         self.dtype = A.dtype if dtype is None else dtype
         return X