MilesCranmer commited on
Commit
3c469fe
1 Parent(s): 21d6b92

Fix issue with warm_start set at init

Browse files
Files changed (1) hide show
  1. pysr/sr.py +3 -1
pysr/sr.py CHANGED
@@ -968,7 +968,9 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
968
  if self.temp_equation_file:
969
  self.equation_file_ = self.tempdir_ / "hall_of_fame.csv"
970
  elif self.equation_file is None:
971
- if self.warm_start and self.equation_file_:
 
 
972
  pass
973
  else:
974
  date_time = datetime.now().strftime("%Y-%m-%d_%H%M%S.%f")[:-3]
 
968
  if self.temp_equation_file:
969
  self.equation_file_ = self.tempdir_ / "hall_of_fame.csv"
970
  elif self.equation_file is None:
971
+ if self.warm_start and (
972
+ hasattr(self, "equation_file_") and self.equation_file_
973
+ ):
974
  pass
975
  else:
976
  date_time = datetime.now().strftime("%Y-%m-%d_%H%M%S.%f")[:-3]