diff --git a/src/pystencils/kernelcreation.py b/src/pystencils/kernelcreation.py index 154cb2307bdaaec0167632d4c77531632b1fe9e8..e31ef2fd42ec80b8c3c123cb20b11fa2a61c47df 100644 --- a/src/pystencils/kernelcreation.py +++ b/src/pystencils/kernelcreation.py @@ -32,7 +32,8 @@ from .backend.transformations import ( SelectFunctions, ) -from .sympyextensions import AssignmentCollection, Assignment +from .simp import AssignmentCollection +from .assignment import Assignment __all__ = ["create_kernel"]