Spaces:
Running
Running
MilesCranmer
commited on
Commit
•
11f524f
1
Parent(s):
6881818
Don't check array if set to None
Browse files- 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
|
|
|
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_
|