MilesCranmer commited on
Commit
0106661
1 Parent(s): 60e8eba

Clear variable names and selection at reset()

Browse files
Files changed (1) hide show
  1. pysr/sr.py +2 -0
pysr/sr.py CHANGED
@@ -903,6 +903,8 @@ class PySRRegressor(BaseEstimator, RegressorMixin):
903
  self.equations = None
904
  self.params_hash = None
905
  self.raw_julia_state = None
 
 
906
 
907
  def _run(self, X, y, weights, variable_names):
908
  global already_ran
 
903
  self.equations = None
904
  self.params_hash = None
905
  self.raw_julia_state = None
906
+ self.variable_names = None
907
+ self.selection = None
908
 
909
  def _run(self, X, y, weights, variable_names):
910
  global already_ran