Let iterator of AssignmentCollection iterate over all_assignments (not only main_assignments)

8479aa27 · Stephan Seitz · 71b8767b · 8479aa27
Commit 8479aa27 authored 5 years ago by Stephan Seitz
--- 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):