Explicitly deleting context object

choderalab · Aug 22, 2022 · 9fad690 · 9fad690
1 parent 24ca6c1
commit 9fad690
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/perses/rjmc/geometry.py b/perses/rjmc/geometry.py
@@ -732,6 +732,9 @@ def _corrected_reduced_potential(self, growth_system_generator, positions, platf
         added_energy_components = compute_potential_components(mod_context)
         print(f"added energy components: {added_energy_components}")
 
+        # Explicitly clean up context memory allocation
+        del mod_context
+
         return modified_reduced_potential_energy