fix: unit vector not removed until end of execution

Loop3D · May 17, 2024 · 0254c73 · 0254c73
1 parent 5e41283
commit 0254c73
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/FoldOptLib/optimisers/axial_surface_optimiser.py b/FoldOptLib/optimisers/axial_surface_optimiser.py
@@ -245,9 +245,7 @@ def optimisation_function(strike_dip):
             angle_difference = ObjectiveFunction[ObjectiveType.ANGLE](
                 predicted_foliation, self.gradient_data
             )
-            # clean up memory
-            del predicted_foliation, unit_vector
-            gc.collect()
+
 
             # If the optimisation type is angle, return the angle difference
             if self.optimisation_type == OptimisationType.ANGLE:
@@ -262,6 +260,10 @@ def optimisation_function(strike_dip):
                     # Calculate the logpdf of the angle difference
                     logpdf = objective_function(angle_difference) + knowledge_function(unit_vector)
                     return logpdf
+
+                # clean up memory
+                del predicted_foliation, unit_vector
+                gc.collect()
 
         return optimisation_function