From 477785f5b700af4a20f79a503730c3d2ebaef088 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Wed, 15 Jan 2020 12:09:21 +0100
Subject: [PATCH] Allow kwargs in fd.Diff.interpolated_access
---
pystencils/fd/derivative.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/pystencils/fd/derivative.py b/pystencils/fd/derivative.py
index 16776a130..02d8f74ee 100644
--- a/pystencils/fd/derivative.py
+++ b/pystencils/fd/derivative.py
@@ -111,15 +111,15 @@ class Diff(sp.Expr):
def __str__(self):
return "D(%s)" % self.arg
- def interpolated_access(self, offset):
+ def interpolated_access(self, offset, **kwargs):
"""Represents an interpolated access on a spatially differentiated field
Args:
offset (Tuple[sympy.Expr]): Absolute position to determine the value of the spatial derivative
"""
from pystencils.interpolation_astnodes import DiffInterpolatorAccess
- assert isinstance(self.argument, Field), "Must be field to enable interpolated accesses"
- return DiffInterpolatorAccess(self.argument.interpolated_access(offset), self.target)
+ assert isinstance(self.arg.field, Field), "Must be field to enable interpolated accesses"
+ return DiffInterpolatorAccess(self.arg.field.interpolated_access(offset, **kwargs).symbol, self.target, *offset)
class DiffOperator(sp.Expr):
--
GitLab