api: fix symbolic coefficient conservation at rebuild

devitocodes · Aug 22, 2023 · fd5b167 · fd5b167
1 parent a38ea46
commit fd5b167
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/devito/types/basic.py b/devito/types/basic.py
@@ -855,7 +855,7 @@ def __str__(self):
 
     __repr__ = __str__
 
-    def _sympystr(self, printer):
+    def _sympystr(self, printer, **kwargs):
         return str(self)
 
     _latex = _sympystr

diff --git a/devito/types/dense.py b/devito/types/dense.py
@@ -68,7 +68,8 @@ def __init_finalize__(self, *args, function=None, **kwargs):
         super(DiscreteFunction, self).__init_finalize__(*args, **kwargs)
 
         # Symbolic (finite difference) coefficients
-        self._coefficients = kwargs.get('coefficients', 'standard')
+        self._coefficients = kwargs.get('coefficients',
+                                        getattr(function, 'coefficients', 'standard'))
         if self._coefficients not in ('standard', 'symbolic'):
             raise ValueError("coefficients must be `standard` or `symbolic`")