diff --git a/src/pystencils/backend/kernelcreation/freeze.py b/src/pystencils/backend/kernelcreation/freeze.py index 4c7b8fb231d707742097aa23110945a6ab34e203..b1bb4cd4a9084310f1eb066ab636eea2a331d91d 100644 --- a/src/pystencils/backend/kernelcreation/freeze.py +++ b/src/pystencils/backend/kernelcreation/freeze.py @@ -63,7 +63,7 @@ from ..ast.vector import PsVecMemAcc from ..constants import PsConstant from ...types import PsNumericType, PsStructType, PsType from ..exceptions import PsInputError -from ..functions import PsMathFunction, MathFunctions, NumericLimitsFunctions +from ..functions import PsMathFunction, MathFunctions from ..exceptions import FreezeError diff --git a/src/pystencils/backend/reduction_op_mapping.py b/src/pystencils/backend/reduction_op_mapping.py index 389c0940ac23d0e2887247fbb33168a3ef1df1e6..a97a496e04e78f4a76be12cae679027a7da56f86 100644 --- a/src/pystencils/backend/reduction_op_mapping.py +++ b/src/pystencils/backend/reduction_op_mapping.py @@ -3,6 +3,7 @@ from .exceptions import PsInternalCompilerError from .functions import PsMathFunction, MathFunctions from ..sympyextensions.reduction import ReductionOp + def reduction_op_to_expr(op: ReductionOp, op1, op2) -> PsExpression: match op: case ReductionOp.Add: