fix gc

MilesCranmer · Aug 12, 2024 · 8a1e363 · 8a1e363
1 parent 184ad23
commit 8a1e363
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pysr/sr.py b/pysr/sr.py
@@ -39,6 +39,7 @@
 from .feature_selection import run_feature_selection
 from .julia_extensions import load_required_packages
 from .julia_helpers import (
+    PythonCall,
     _escape_filename,
     _load_cluster_manager,
     jl_array,
@@ -1885,6 +1886,7 @@ def _run(
         else:
             jl_y_variable_names = None
 
+        PythonCall.GC.disable()
         out = SymbolicRegression.equation_search(
             jl_X,
             jl_y,
@@ -1911,6 +1913,7 @@ def _run(
             progress=progress and self.verbosity > 0 and len(y.shape) == 1,
             verbosity=int(self.verbosity),
         )
+        PythonCall.GC.enable()
 
         self.julia_state_stream_ = jl_serialize(out)