From 66ce43954c585e5d576f895ab0d60f62196db813 Mon Sep 17 00:00:00 2001
From: zy69guqi <richard.angersbach@fau.de>
Date: Wed, 15 Jan 2025 17:19:09 +0100
Subject: [PATCH] Minor import fixes
---
src/pystencils/backend/kernelcreation/freeze.py | 9 ++++-----
src/pystencils/codegen/properties.py | 2 +-
src/pystencils/sympyextensions/__init__.py | 2 --
3 files changed, 5 insertions(+), 8 deletions(-)
diff --git a/src/pystencils/backend/kernelcreation/freeze.py b/src/pystencils/backend/kernelcreation/freeze.py
index 0d1ce72e1..7316e2f9f 100644
--- a/src/pystencils/backend/kernelcreation/freeze.py
+++ b/src/pystencils/backend/kernelcreation/freeze.py
@@ -62,11 +62,10 @@ from ..ast.vector import PsVecMemAcc
from ..constants import PsConstant
from ...types import PsNumericType, PsStructType, PsType
from ..exceptions import PsInputError
-from ..functions import PsMathFunction, MathFunctions
+from ..functions import PsMathFunction, MathFunctions, NumericLimitsFunctions
from ..exceptions import FreezeError
-import backend.functions
-from codegen.properties import ReductionSymbolProperty
+from ...codegen.properties import ReductionSymbolProperty
ExprLike = (
@@ -207,10 +206,10 @@ class FreezeExpressions:
# TODO: unsure if sp.Min & sp.Max are mapped by map_Min/map_Max afterwards
case "min":
op = sp.Min
- init_val = backend.functions.NumericLimitsFunctions("min")
+ init_val = NumericLimitsFunctions("min")
case "max":
op = sp.Max
- init_val = backend.functions.NumericLimitsFunctions("max")
+ init_val = NumericLimitsFunctions("max")
case _:
raise FreezeError(f"Unsupported reduced assignment: {expr.op}.")
diff --git a/src/pystencils/codegen/properties.py b/src/pystencils/codegen/properties.py
index 5578d2408..cc4ff4101 100644
--- a/src/pystencils/codegen/properties.py
+++ b/src/pystencils/codegen/properties.py
@@ -3,7 +3,7 @@ from dataclasses import dataclass
from ..field import Field
-from backend.ast.expressions import PsExpression
+from ..backend.ast.expressions import PsExpression
@dataclass(frozen=True)
diff --git a/src/pystencils/sympyextensions/__init__.py b/src/pystencils/sympyextensions/__init__.py
index 6ab24e936..7431416c9 100644
--- a/src/pystencils/sympyextensions/__init__.py
+++ b/src/pystencils/sympyextensions/__init__.py
@@ -1,7 +1,6 @@
from .astnodes import ConditionalFieldAccess
from .typed_sympy import TypedSymbol, CastFunc
from .pointers import mem_acc
-from .reduction import reduced_assign
from .math import (
prod,
@@ -34,7 +33,6 @@ from .math import (
__all__ = [
"ConditionalFieldAccess",
- "reduced_assign",
"TypedSymbol",
"CastFunc",
"mem_acc",
--
GitLab