Attempt to fix regularity recombination when array has size 0

DedalusProject · Jul 18, 2024 · e3cab89 · e3cab89
1 parent 51ca757
commit e3cab89
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dedalus/core/basis.py b/dedalus/core/basis.py
@@ -3600,7 +3600,7 @@ def forward_regularity_recombination(self, tensorsig, axis, gdata, ell_maps):
             Q = self.radial_recombinations(tensorsig, ell_list)
             # Flatten tensor axes
             shape = gdata.shape
-            temp = gdata.reshape((-1,)+shape[rank:])
+            temp = gdata.reshape((np.prod(shape[:rank]),)+shape[rank:])
             slices = [slice(None) for i in range(temp.ndim)]
             # Apply Q transformations for each ell to flattened tensor data
             for ell, m_ind, ell_ind in ell_maps:
@@ -3617,7 +3617,7 @@ def backward_regularity_recombination(self, tensorsig, axis, gdata, ell_maps):
             Q = self.radial_recombinations(tensorsig, ell_list)
             # Flatten tensor axes
             shape = gdata.shape
-            temp = gdata.reshape((-1,)+shape[rank:])
+            temp = gdata.reshape((np.prod(shape[:rank]),)+shape[rank:])
             slices = [slice(None) for i in range(temp.ndim)]
             # Apply Q transformations for each ell to flattened tensor data
             for ell, m_ind, ell_ind in ell_maps: