From 3daaa5e5a1f92a5482cb838cd791ed062dae1398 Mon Sep 17 00:00:00 2001
From: zy69guqi <richard.angersbach@fau.de>
Date: Tue, 28 Jan 2025 12:47:15 +0100
Subject: [PATCH] Fix typecheck
---
src/pystencils/sympyextensions/__init__.py | 2 ++
src/pystencils/sympyextensions/binop_mapping.py | 6 +++---
2 files changed, 5 insertions(+), 3 deletions(-)
diff --git a/src/pystencils/sympyextensions/__init__.py b/src/pystencils/sympyextensions/__init__.py
index 6ab24e936..8d832ba2a 100644
--- a/src/pystencils/sympyextensions/__init__.py
+++ b/src/pystencils/sympyextensions/__init__.py
@@ -2,6 +2,7 @@ from .astnodes import ConditionalFieldAccess
from .typed_sympy import TypedSymbol, CastFunc
from .pointers import mem_acc
from .reduction import reduced_assign
+from .binop_mapping import binop_str_to_expr
from .math import (
prod,
@@ -35,6 +36,7 @@ from .math import (
__all__ = [
"ConditionalFieldAccess",
"reduced_assign",
+ "binop_str_to_expr",
"TypedSymbol",
"CastFunc",
"mem_acc",
diff --git a/src/pystencils/sympyextensions/binop_mapping.py b/src/pystencils/sympyextensions/binop_mapping.py
index 1cb2a3ab5..04cfb6107 100644
--- a/src/pystencils/sympyextensions/binop_mapping.py
+++ b/src/pystencils/sympyextensions/binop_mapping.py
@@ -1,8 +1,8 @@
from operator import truediv, mul, sub, add
-from src.pystencils.backend.ast.expressions import PsCall, PsExpression
-from src.pystencils.backend.exceptions import FreezeError
-from src.pystencils.backend.functions import MathFunctions, PsMathFunction
+from ..backend.ast.expressions import PsCall, PsExpression
+from ..backend.exceptions import FreezeError
+from ..backend.functions import MathFunctions, PsMathFunction
_available_operator_interface: set[str] = {'+', '-', '*', '/'}
--
GitLab