Eliminate deprecation warning for OpenMM 7.6 (#12)

SSAGESLabs · Oct 16, 2021 · 2ec5ff9 · 2ec5ff9
2 parents 478a11b + 5445b07
commit 2ec5ff9
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -2,3 +2,4 @@ FROM ssages/pysages-hoomd:latest
 
 COPY . openmm-dlext
 RUN cd openmm-dlext && mkdir build && cd build && cmake .. && make install
+RUN python -c "import openmm_dlext"
diff --git a/wrappers/python/DLExtForce.i b/wrappers/python/DLExtForce.i
@@ -1,6 +1,15 @@
 %module dlpack_extension_swig
 
-%import(module = "simtk.openmm") "OpenMMForce.i"
+%pythoncode
+%{
+from importlib import import_module
+try:
+    openmm = import_module("openmm")
+except:
+    openmm = import_module("simtk.openmm")
+%}
+
+%import(module = "openmm") "OpenMMForce.i"
 %include "std_string.i"
 
 %{
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2,3 +2,4 @@ FROM ssages/pysages-hoomd:latest

		COPY . openmm-dlext
		RUN cd openmm-dlext && mkdir build && cd build && cmake .. && make install
		RUN python -c "import openmm_dlext"