From 7d1a1de8954c8d7d5fb9cf6f6004066a36aef022 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Wed, 29 Jan 2025 13:16:14 +0100
Subject: [PATCH] fix CastFunc imports

---
 doc/notebooks/demo_thermalized_lbm.ipynb         | 9 +++------
 src/lbmpy/boundaries/boundaryconditions.py       | 2 +-
 src/lbmpy/phasefield_allen_cahn/contact_angle.py | 2 +-
 3 files changed, 5 insertions(+), 8 deletions(-)

diff --git a/doc/notebooks/demo_thermalized_lbm.ipynb b/doc/notebooks/demo_thermalized_lbm.ipynb
index 9c0d9579..b9f759bd 100644
--- a/doc/notebooks/demo_thermalized_lbm.ipynb
+++ b/doc/notebooks/demo_thermalized_lbm.ipynb
@@ -183,7 +183,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -216,7 +216,7 @@
     }
    ],
    "source": [
-    "from pystencils.sympyextensions import CastFunc\n",
+    "from pystencils.sympyextensions.typed_sympy import CastFunc\n",
     "\n",
     "dh = ps.create_data_handling(domain_size=(80, 80))\n",
     "\n",
@@ -347,11 +347,8 @@
   }
  ],
  "metadata": {
-  "interpreter": {
-   "hash": "ca06c80c4febc35b85e85156d391051f9f4a8895eee3f708eb1f33a09d8697a0"
-  },
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": ".venv",
    "language": "python",
    "name": "python3"
   },
diff --git a/src/lbmpy/boundaries/boundaryconditions.py b/src/lbmpy/boundaries/boundaryconditions.py
index 37304460..3021450d 100644
--- a/src/lbmpy/boundaries/boundaryconditions.py
+++ b/src/lbmpy/boundaries/boundaryconditions.py
@@ -3,7 +3,7 @@ from enum import Enum, auto
 from warnings import warn
 
 from pystencils import Assignment, AssignmentCollection, Field, TypedSymbol
-from pystencils.sympyextensions import CastFunc
+from pystencils.sympyextensions.typed_sympy import CastFunc
 from pystencils.stencil import offset_to_direction_string, direction_string_to_offset, inverse_direction
 from pystencils.sympyextensions import get_symmetric_part, simplify_by_equality, scalar_product
 
diff --git a/src/lbmpy/phasefield_allen_cahn/contact_angle.py b/src/lbmpy/phasefield_allen_cahn/contact_angle.py
index ef613194..f7ed4b35 100644
--- a/src/lbmpy/phasefield_allen_cahn/contact_angle.py
+++ b/src/lbmpy/phasefield_allen_cahn/contact_angle.py
@@ -12,7 +12,7 @@ from pystencils.boundaries.boundaryhandling import BoundaryOffsetInfo
 from pystencils.boundaries.boundaryconditions import Boundary
 
 from pystencils import TypedSymbol
-from pystencils.sympyextensions import CastFunc
+from pystencils.sympyextensions.typed_sympy import CastFunc
 
 
 class ContactAngle(Boundary):
-- 
GitLab