MilesCranmer commited on
Commit
85371bb
1 Parent(s): 179fef6

Allow pickling without equations_ stored

Browse files
Files changed (1) hide show
  1. pysr/sr.py +3 -1
pysr/sr.py CHANGED
@@ -883,7 +883,9 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
883
  key: None if key == "raw_julia_state_" else value
884
  for key, value in state.items()
885
  }
886
- if "equations_" in pickled_state:
 
 
887
  pickled_state["output_torch_format"] = False
888
  pickled_state["output_jax_format"] = False
889
  if self.nout_ == 1:
 
883
  key: None if key == "raw_julia_state_" else value
884
  for key, value in state.items()
885
  }
886
+ if ("equations_" in pickled_state) and (
887
+ pickled_state["equations_"] is not None
888
+ ):
889
  pickled_state["output_torch_format"] = False
890
  pickled_state["output_jax_format"] = False
891
  if self.nout_ == 1: