From 8479aa27486f408c7e0d9099d966d79b05ffd205 Mon Sep 17 00:00:00 2001 From: Stephan Seitz <stephan.seitz@fau.de> Date: Sun, 18 Aug 2019 11:23:50 +0200 Subject: [PATCH] Let iterator of AssignmentCollection iterate over all_assignments (not only main_assignments) --- pystencils/simp/assignment_collection.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pystencils/simp/assignment_collection.py b/pystencils/simp/assignment_collection.py index fa437a5d0..1e9a4c47e 100644 --- a/pystencils/simp/assignment_collection.py +++ b/pystencils/simp/assignment_collection.py @@ -347,7 +347,7 @@ class AssignmentCollection: return result def __iter__(self): - return self.main_assignments.__iter__() + return self.all_assignments.__iter__() @property def main_assignments_dict(self): -- GitLab