Patch integrate and average to properly expand cartesian coordsystems

DedalusProject · Feb 16, 2022 · 336b993 · 336b993
1 parent 18c0cdf
commit 336b993
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/dedalus/core/operators.py b/dedalus/core/operators.py
@@ -1095,6 +1095,9 @@ def _preprocess_args(cls, operand, coord=None):
         # Integrate over all operand bases by default
         if coord is None:
             coord = [basis.coordsystem for basis in operand.domain.bases]
+        # Split Cartesian coordinates
+        if isinstance(coord, coords.CartesianCoordinates):
+            coord = coord.coords
         # Recurse over multiple coordinates
         if isinstance(coord, (tuple, list)):
             if len(coord) > 1:
@@ -1162,6 +1165,9 @@ def _preprocess_args(cls, operand, coord=None):
         # Average over all operand bases by default
         if coord is None:
             coord = [basis.coordsystem for basis in operand.domain.bases]
+        # Split Cartesian coordinates
+        if isinstance(coord, coords.CartesianCoordinates):
+            coord = coord.coords
         # Recurse over multiple coordinates
         if isinstance(coord, (tuple, list)):
             if len(coord) > 1: