Fix memory error in PythonForce (#5182)

421eb42a · Peter Eastman · GitHub · e688b86e · 421eb42a
Unverified Commit 421eb42a authored Jan 13, 2026 by Peter Eastman Committed by GitHub Jan 13, 2026
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

wrappers/python/src/swig_doxygen/swig_lib/python/pythonforce.i ...ers/python/src/swig_doxygen/swig_lib/python/pythonforce.i +4 -0

No files found.
--- a/wrappers/python/src/swig_doxygen/swig_lib/python/pythonforce.i
+++ b/wrappers/python/src/swig_doxygen/swig_lib/python/pythonforce.i
@@ -12,6 +12,10 @@ namespace OpenMM {
    class ComputationWrapper : public PythonForceComputation {
    public:
        ComputationWrapper(PyObject* computation) : computation(computation) {
+            Py_INCREF(computation);
+        }
+        ~ComputationWrapper() {
+            Py_XDECREF(computation);
        }
        void compute(const State& state, double& energy, void* forces, bool forcesAreDouble) const {
            PyGILState_STATE gstate;