From 768169b39aa00a490ab5f5d6c34ff96d39efb9f8 Mon Sep 17 00:00:00 2001 From: Johnnie Gray Date: Tue, 28 Jan 2025 18:08:57 -0800 Subject: [PATCH] diis: reshape with tuple always --- quimb/experimental/belief_propagation/diis.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/quimb/experimental/belief_propagation/diis.py b/quimb/experimental/belief_propagation/diis.py index 17502303..9544501e 100644 --- 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)