imports, fix

wilson-labs · Sep 11, 2023 · b77995c · b77995c
1 parent cb82fea
commit b77995c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cola/linalg/inv.py b/cola/linalg/inv.py
@@ -2,8 +2,8 @@
 from plum import dispatch
 from cola.ops import LinearOperator
 from cola.ops import Diagonal, Permutation
-from cola.ops import Identity
-from cola.ops import ScalarMul
+from cola.ops import Identity, Dense
+from cola.ops import ScalarMul, Sum
 from cola.ops import BlockDiag, Triangular
 from cola.ops import Kronecker, Product
 from cola.algorithms.cg import cg
@@ -180,4 +180,4 @@ def inv(A: Sum[Product[Dense,Dense], Diagonal], **kwargs):
 
 @dispatch
 def inv(A: Sum[Diagonal, Product[Dense,Dense]], **kwargs):
-    return inv(Product(*C.Ms[1].Ms) + C.Ms[0])
+    return inv(Product(*A.Ms[1].Ms) + A.Ms[0])