From 223a1051bf50eb1ffe97bd3b0b51349c7d285ca1 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Fri, 25 Mar 2022 09:14:17 +0100
Subject: [PATCH] Fix: AssignmentCollection's `new_filtered` now also finds
dependencies in 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 f493b08e9..07d29f3dc 100644
--- a/pystencils/simp/assignment_collection.py
+++ b/pystencils/simp/assignment_collection.py
@@ -303,7 +303,7 @@ class AssignmentCollection:
if eq.lhs in symbols_to_extract:
new_assignments.append(eq)
- new_sub_expr = [eq for eq in self.subexpressions
+ new_sub_expr = [eq for eq in self.all_assignments
if eq.lhs in dependent_symbols and eq.lhs not in symbols_to_extract]
return self.copy(new_assignments, new_sub_expr)
--
GitLab