diff --git a/src/pystencils_autodiff/framework_integration/astnodes.py b/src/pystencils_autodiff/framework_integration/astnodes.py index 7ef512120ab5af7aab12d8a6cfd88f0cb420c2a1..ed16a4dbb7054750be640faa70645e1a854cd6db 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