Merge pull request #22 from SSAGESLabs/pz/pybind11

Import hoomd and hoomd.md modules
SSAGESLabs · Sep 6, 2023 · 9deea21 · 9deea21
2 parents 8d758ea + f75b170
commit 9deea21
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/hoomd_dlext.cc b/python/hoomd_dlext.cc
@@ -41,6 +41,9 @@ void export_PySampler(py::module m)
     py::class_<HalfStepHook, PyHalfStepHook, SPtr<HalfStepHook>>(m, "HalfStepHook")
         .def(py::init<>())
         .def("update", &HalfStepHook::update);
+#else
+    py::module_::import("hoomd");
+    py::module_::import("hoomd.md");
 #endif
 
     py::class_<PySampler, SPtr<PySampler>, HalfStepHook>(m, "DLExtSampler")