From b77720edbee17f7b4a3c9d21e169b7a03d8b69a8 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Thu, 16 Jan 2025 10:37:05 +0100
Subject: [PATCH] adapt to changes in pystencils API

---
 conftest.py                    | 2 +-
 src/lbmpy/creationfunctions.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/conftest.py b/conftest.py
index 23ed4d49..b25dbeca 100644
--- a/conftest.py
+++ b/conftest.py
@@ -16,7 +16,7 @@ from lbmpy._compat import IS_PYSTENCILS_2
 # Trigger config file reading / creation once - to avoid race conditions when multiple instances are creating it
 # at the same time
 if IS_PYSTENCILS_2:
-    from pystencils.backend.jit import legacy_cpu
+    from pystencils.jit import legacy_cpu
 else:
     from pystencils.cpu import cpujit
 
diff --git a/src/lbmpy/creationfunctions.py b/src/lbmpy/creationfunctions.py
index dcdbdd1c..1883a1e2 100644
--- a/src/lbmpy/creationfunctions.py
+++ b/src/lbmpy/creationfunctions.py
@@ -93,7 +93,7 @@ from lbmpy.methods.abstractlbmethod import LbmCollisionRule, AbstractLbMethod
 from lbmpy.methods.cumulantbased import CumulantBasedLbMethod
 
 if IS_PYSTENCILS_2:
-    from pystencils import KernelFunction
+    from pystencils import Kernel as KernelFunction
 else:
     from pystencils.astnodes import KernelFunction
 
-- 
GitLab