Skip to content

Commit 72cd436

Browse files
committed
Remove _validate_pandas_input
1 parent dbec9f3 commit 72cd436

File tree

1 file changed

+1
-12
lines changed

1 file changed

+1
-12
lines changed

boruta/boruta_py.py

Lines changed: 1 addition & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -283,14 +283,6 @@ def fit_transform(self, X, y=None, **fit_params):
283283
return_df = fit_params.pop("return_df", None)
284284
return self.fit(X, y, **fit_params).transform(X, weak=weak, return_df=return_df)
285285

286-
def _validate_pandas_input(self, arg):
287-
try:
288-
return arg.values
289-
except AttributeError:
290-
raise ValueError(
291-
"input needs to be a numpy array or pandas data frame."
292-
)
293-
294286
def _fit(self, X, y):
295287
# check input params
296288
self._check_params(X, y)
@@ -301,10 +293,7 @@ def _fit(self, X, y):
301293
else:
302294
self.feature_names_in_ = None
303295

304-
if not isinstance(X, np.ndarray):
305-
X = self._validate_pandas_input(X)
306-
if not isinstance(y, np.ndarray):
307-
y = self._validate_pandas_input(y)
296+
X, y = check_X_y(X, y, accept_sparse=False, ensure_2d=True, dtype=None, estimator=self)
308297

309298
self.n_features_in_ = X.shape[1]
310299

0 commit comments

Comments
 (0)