From f4ae2f58ac1557d140f9db825248b537001641b7 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 24 Oct 2019 16:31:42 +0200
Subject: [PATCH] Improve __repr__ of JinjaCppFile

---
 src/pystencils_autodiff/framework_integration/astnodes.py | 2 +-
 src/pystencils_autodiff/transformations.py                | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/pystencils_autodiff/framework_integration/astnodes.py b/src/pystencils_autodiff/framework_integration/astnodes.py
index fbfed82..7ef5121 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 5ba38b8..2ffdaf2 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
-- 
GitLab