From 666fa8d6c68a4e4dd4f35ea80876a7b4f8f425b3 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Wed, 16 Mar 2022 16:27:24 +0100
Subject: [PATCH] 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