Skip to content

Commit 4348a2f

Browse files
committed
Fix SphereEllProduct dispatch
1 parent dc4decb commit 4348a2f

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

dedalus/core/operators.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2654,12 +2654,22 @@ def operate(self, out):
26542654
comp_out += symbols * comp_in # TEMPORARY
26552655

26562656

2657-
class SphereEllProduct(SeparableSphereOperator):
2657+
class SphereEllProduct(SeparableSphereOperator, metaclass=MultiClass):
26582658

26592659
name = "SphereEllProduct"
26602660
complex_operator = False
26612661
subaxis_dependence = [False, True]
26622662

2663+
@classmethod
2664+
def _preprocess_args(cls, operand, coordsys, ell_r_func, out=None):
2665+
if operand == 0:
2666+
raise SkipDispatchException(output=0)
2667+
return [operand, coordsys, ell_r_func], {'out': out}
2668+
2669+
@classmethod
2670+
def _check_args(cls, operand, coordsys, ell_r_func, out=None):
2671+
return True
2672+
26632673
def __init__(self, operand, coordsys, ell_r_func, out=None):
26642674
super().__init__(operand, out=out)
26652675
self.ell_r_func = ell_r_func

0 commit comments

Comments
 (0)