MilesCranmer commited on
Commit
e5b4869
·
1 Parent(s): ccf71e9

Call `refresh` in load function

Browse files
Files changed (1) hide show
  1. pysr/sr.py +1 -2
pysr/sr.py CHANGED
@@ -2090,7 +2090,6 @@ def load(
2090
  **pysr_kwargs,
2091
  )
2092
 
2093
- model.equation_file_ = equation_file
2094
  model.nout_ = nout
2095
  model.n_features_in_ = n_features_in
2096
 
@@ -2105,6 +2104,6 @@ def load(
2105
  else:
2106
  model.selection_mask_ = selection_mask
2107
 
2108
- model.refresh()
2109
 
2110
  return model
 
2090
  **pysr_kwargs,
2091
  )
2092
 
 
2093
  model.nout_ = nout
2094
  model.n_features_in_ = n_features_in
2095
 
 
2104
  else:
2105
  model.selection_mask_ = selection_mask
2106
 
2107
+ model.refresh(checkpoint_file=equation_file)
2108
 
2109
  return model