Spaces:

MilesCranmer
/

PySR

Running

MilesCranmer commited on Dec 30, 2022

Commit

7c8d588

•

1 Parent(s): cc21173

Fix bug with `cluster_manager` not loading

Files changed (1) hide show

pysr/sr.py CHANGED Viewed

@@ -1493,7 +1493,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
         Main = init_julia(self.julia_project, julia_kwargs=julia_kwargs)
         if cluster_manager is not None:
-            cluster_manager = _load_cluster_manager(cluster_manager)
         if self.update:
             _, is_shared = _process_julia_project(self.julia_project)

         Main = init_julia(self.julia_project, julia_kwargs=julia_kwargs)
         if cluster_manager is not None:
+            cluster_manager = _load_cluster_manager(Main, cluster_manager)
         if self.update:
             _, is_shared = _process_julia_project(self.julia_project)