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