Spaces:

MilesCranmer
/

PySR

Running

MilesCranmer commited on Jul 20, 2022

Commit

e5b4869

1 Parent(s): ccf71e9

Call `refresh` in load function

Files changed (1) hide show

pysr/sr.py CHANGED Viewed

@@ -2090,7 +2090,6 @@ def load(
         **pysr_kwargs,
     )
-    model.equation_file_ = equation_file
     model.nout_ = nout
     model.n_features_in_ = n_features_in
@@ -2105,6 +2104,6 @@ def load(
     else:
         model.selection_mask_ = selection_mask
-    model.refresh()
     return model

         **pysr_kwargs,
     )
     model.nout_ = nout
     model.n_features_in_ = n_features_in
     else:
         model.selection_mask_ = selection_mask
+    model.refresh(checkpoint_file=equation_file)
     return model