diis: reshape with tuple always

jcmgray · Jan 29, 2025 · 768169b · 768169b
1 parent 7e6108a
commit 768169b
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
@@ -75,7 +75,7 @@ def pack(self, tree):
         def extractor(x):
             if isinstance(x, Tensor):
                 x = x.data
-            arrays.append(self._reshape(x, -1))
+            arrays.append(self._reshape(x, (-1,)))
 
         arrays = []
         tree_apply(extractor, tree)