From f54fa321869682e71edd3f2828725104de9abe36 Mon Sep 17 00:00:00 2001
From: zy69guqi <richard.angersbach@fau.de>
Date: Wed, 15 Jan 2025 11:36:53 +0100
Subject: [PATCH] Fix relative module imports for newly introduced
 sympyextensions for reductions

---
 src/pystencils/backend/kernelcreation/freeze.py | 2 +-
 src/pystencils/simp/assignment_collection.py    | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/pystencils/backend/kernelcreation/freeze.py b/src/pystencils/backend/kernelcreation/freeze.py
index 65be23065..4d75f1ca6 100644
--- a/src/pystencils/backend/kernelcreation/freeze.py
+++ b/src/pystencils/backend/kernelcreation/freeze.py
@@ -1,4 +1,3 @@
-from sympyextensions.reduction import ReducedAssignment
 from typing import overload, cast, Any
 from functools import reduce
 from operator import add, mul, sub, truediv
@@ -16,6 +15,7 @@ from ...sympyextensions import (
 )
 from ...sympyextensions.typed_sympy import TypedSymbol, CastFunc, DynamicType
 from ...sympyextensions.pointers import AddressOf, mem_acc
+from ...sympyextensions.reduction import ReducedAssignment
 from ...field import Field, FieldType
 
 from .context import KernelCreationContext
diff --git a/src/pystencils/simp/assignment_collection.py b/src/pystencils/simp/assignment_collection.py
index 4de3e8dc6..212dbf751 100644
--- a/src/pystencils/simp/assignment_collection.py
+++ b/src/pystencils/simp/assignment_collection.py
@@ -1,8 +1,6 @@
 import itertools
 from copy import copy
 
-from sympyextensions import reduced_assign
-from sympyextensions.reduction import ReducedAssignment
 from typing import Any, Dict, Iterable, Iterator, List, Optional, Sequence, Set, Union
 
 import sympy as sp
@@ -11,6 +9,8 @@ import pystencils
 from ..assignment import Assignment
 from .simplifications import (sort_assignments_topologically, transform_lhs_and_rhs, transform_rhs)
 from ..sympyextensions import count_operations, fast_subs
+from ..sympyextensions import reduced_assign
+from ..sympyextensions.reduction import ReducedAssignment
 
 
 class AssignmentCollection:
-- 
GitLab