diff --git a/src/pystencils_autodiff/backends/python_bindings.py b/src/pystencils_autodiff/backends/python_bindings.py index 7410d943579c344994ca910d3fef2c571434e2ea..fd6f02bd6b0d11451a007a8b75db1cfa9a0605c7 100644 --- a/src/pystencils_autodiff/backends/python_bindings.py +++ b/src/pystencils_autodiff/backends/python_bindings.py @@ -13,6 +13,7 @@ import jinja2 import stringcase from pystencils.astnodes import KernelFunction +from pystencils.backends.cbackend import CustomCodeNode from pystencils_autodiff.framework_integration.astnodes import JinjaCppFile @@ -158,7 +159,7 @@ class PybindFunctionWrapping(JinjaCppFile): """m.def("{{ python_name }}", &{{ cpp_name }}{% for p in parameters -%}, "{{ p }}"_a{% endfor %});""" # noqa ) - required_global_declarations = ["using namespace pybind11::literals;"] + required_global_declarations = [CustomCodeNode("using namespace pybind11::literals;", (), ())] headers = ['<pybind11/pybind11.h>', '<pybind11/stl.h>']