fix vectorizer bug

jcmgray · Jan 29, 2025 · 80360c8 · 80360c8
1 parent 1976ddd
commit 80360c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/quimb/experimental/belief_propagation/diis.py b/quimb/experimental/belief_propagation/diis.py
@@ -49,7 +49,7 @@ def extracter(x):
                 info = x
             else:
                 array = x
-                shape = ar.do("shape", x.shape, like=self.backend)
+                shape = ar.do("shape", x, like=self.backend)
                 size = ar.do("size", x, like=self.backend)
                 info = ArrayInfo(shape, size)