adapt to changes in pystencils API

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
 

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