From 1972dcf8d1f7598d66dde5cdd73174ae1ef917a5 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 24 Oct 2019 18:07:38 +0200
Subject: [PATCH] Use iterables in JinjaCppFile
---
src/pystencils_autodiff/framework_integration/astnodes.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/pystencils_autodiff/framework_integration/astnodes.py b/src/pystencils_autodiff/framework_integration/astnodes.py
index 7ef5121..ed16a4d 100644
--- a/src/pystencils_autodiff/framework_integration/astnodes.py
+++ b/src/pystencils_autodiff/framework_integration/astnodes.py
@@ -234,7 +234,7 @@ class JinjaCppFile(Node):
def args(self):
"""Returns all arguments/children of this node."""
ast_nodes = [a for a in self.ast_dict.values() if isinstance(a, (Node, str))]
- iterables_of_ast_nodes = [a for a in self.ast_dict.values() if not isinstance(a, (Node, str))]
+ iterables_of_ast_nodes = [a for a in self.ast_dict.values() if isinstance(a, Iterable)]
return ast_nodes + list(itertools.chain.from_iterable(iterables_of_ast_nodes))
@property
--
GitLab