api: fix symbolic coefficient conservation at rebuild

devitocodes · Aug 23, 2023 · 25c856a · 25c856a
1 parent 9b63f91
commit 25c856a
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 2 deletions.
diff --git a/devito/mpatches/.__init__.py.swp b/devito/mpatches/.__init__.py.swp
diff --git a/devito/mpatches/.rationaltools.py.swp b/devito/mpatches/.rationaltools.py.swp
diff --git a/devito/types/basic.py b/devito/types/basic.py
@@ -6,6 +6,7 @@
 
 import numpy as np
 import sympy
+
 from sympy.core.assumptions import _assume_rules
 from cached_property import cached_property
 
@@ -855,11 +856,14 @@ def __str__(self):
 
     __repr__ = __str__
 
-    def _sympystr(self, printer):
+    def _sympystr(self, printer, **kwargs):
         return str(self)
 
     _latex = _sympystr
 
+    def _pretty(self, printer, **kwargs):
+        return printer._print_Function(self, func_name=self.name)
+
     def __eq__(self, other):
         try:
             return (self.function is other.function and

diff --git a/devito/types/dense.py b/devito/types/dense.py
@@ -57,7 +57,7 @@ class DiscreteFunction(AbstractFunction, ArgProvider, Differentiable):
     The type of the underlying data object.
     """
 
-    __rkwargs__ = AbstractFunction.__rkwargs__ + ('staggered',)
+    __rkwargs__ = AbstractFunction.__rkwargs__ + ('staggered', 'coefficients')
 
     def __init_finalize__(self, *args, function=None, **kwargs):
         # Staggering metadata