diff --git a/src/pystencils_autodiff/framework_integration/astnodes.py b/src/pystencils_autodiff/framework_integration/astnodes.py index fbfed8290080cce9bcb7302fa49343bd68e25034..7ef512120ab5af7aab12d8a6cfd88f0cb420c2a1 100644 --- a/src/pystencils_autodiff/framework_integration/astnodes.py +++ b/src/pystencils_autodiff/framework_integration/astnodes.py @@ -282,4 +282,4 @@ class JinjaCppFile(Node): return self.TEMPLATE.render(render_dict) def __repr__(self): - return self.TEMPLATE.render(self.ast_dict) + return f'{str(self.__class__)}:\n {self.TEMPLATE.render(self.ast_dict)}' diff --git a/src/pystencils_autodiff/transformations.py b/src/pystencils_autodiff/transformations.py index 5ba38b8f1be7dfc9d3e39c1705c9737dd9c3fa32..2ffdaf2ba718b2b1acdcbf0524cd0298ea82262a 100644 --- a/src/pystencils_autodiff/transformations.py +++ b/src/pystencils_autodiff/transformations.py @@ -8,7 +8,6 @@ """ import itertools - import sympy as sp import pystencils