Fix SphereEllProduct dispatch

DedalusProject · Apr 1, 2022 · 4348a2f · 4348a2f
1 parent dc4decb
commit 4348a2f
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/dedalus/core/operators.py b/dedalus/core/operators.py
@@ -2654,12 +2654,22 @@ def operate(self, out):
                     comp_out += symbols * comp_in # TEMPORARY
 
 
-class SphereEllProduct(SeparableSphereOperator):
+class SphereEllProduct(SeparableSphereOperator, metaclass=MultiClass):
 
     name = "SphereEllProduct"
     complex_operator = False
     subaxis_dependence = [False, True]
 
+    @classmethod
+    def _preprocess_args(cls, operand, coordsys, ell_r_func, out=None):
+        if operand == 0:
+            raise SkipDispatchException(output=0)
+        return [operand, coordsys, ell_r_func], {'out': out}
+
+    @classmethod
+    def _check_args(cls, operand, coordsys, ell_r_func, out=None):
+        return True
+
     def __init__(self, operand, coordsys, ell_r_func, out=None):
         super().__init__(operand, out=out)
         self.ell_r_func = ell_r_func