MilesCranmer commited on
Commit
11f524f
1 Parent(s): 6881818

Don't check array if set to None

Browse files
Files changed (1) hide show
  1. pysr/sr.py +2 -1
pysr/sr.py CHANGED
@@ -1061,7 +1061,8 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
1061
 
1062
  # Data validation and feature name fetching via sklearn
1063
  # This method sets the n_features_in_ attribute
1064
- Xresampled = check_array(Xresampled)
 
1065
  X, y = self._validate_data(X=X, y=y, reset=True, multi_output=True)
1066
  self.feature_names_in_ = _check_feature_names_in(self, variable_names)
1067
  variable_names = self.feature_names_in_
 
1061
 
1062
  # Data validation and feature name fetching via sklearn
1063
  # This method sets the n_features_in_ attribute
1064
+ if Xresampled is not None:
1065
+ Xresampled = check_array(Xresampled)
1066
  X, y = self._validate_data(X=X, y=y, reset=True, multi_output=True)
1067
  self.feature_names_in_ = _check_feature_names_in(self, variable_names)
1068
  variable_names = self.feature_names_in_