api: fix symbolic coefficient conservation at rebuild

Author: mloubout

devito/mpatches/.__init__.py.swp

@@ -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

devito/mpatches/.rationaltools.py.swp

@@ -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