diff --git a/src/pystencils_autodiff/backends/python_bindings.py b/src/pystencils_autodiff/backends/python_bindings.py index d5c2f710aa8a154e9f0a83c49c562ca263321456..140fbf000ee06a16b3a858df3fe02487e08d4f25 100644 --- a/src/pystencils_autodiff/backends/python_bindings.py +++ b/src/pystencils_autodiff/backends/python_bindings.py @@ -17,7 +17,7 @@ from pystencils_autodiff.framework_integration.astnodes import JinjaCppFile class PybindPythonBindings(JinjaCppFile): - TEMPLATE = jinja2.Template("""PYBIND11_MODULE("{{ module_name }}", m) + TEMPLATE = jinja2.Template("""PYBIND11_MODULE({{ module_name }}, m) { {% for ast_node in module_contents -%} {{ ast_node | indent(3,true) }}