Skip to content

Commit 5e1032c

Browse files
committed
sympy: Mathias patch
1 parent d06773a commit 5e1032c

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

devito/finite_differences/derivative.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,11 @@ def _process_kwargs(cls, expr, *dims, **kwargs):
138138
fd_orders = kwargs.get('fd_order')
139139
deriv_orders = kwargs.get('deriv_order')
140140
if len(dims) == 1:
141-
dims = tuple([dims[0]]*max(1, deriv_orders[0]))
141+
if isinstance(dims[0], Iterable):
142+
assert dims[0][1] == deriv_orders[0]
143+
dims = tuple([dims[0][0]]*max(1, deriv_orders[0]))
144+
else:
145+
dims = tuple([dims[0]]*max(1, deriv_orders[0]))
142146
variable_count = [sympy.Tuple(s, dims.count(s))
143147
for s in filter_ordered(dims)]
144148
return dims, deriv_orders, fd_orders, variable_count

0 commit comments

Comments
 (0)