Skip to content

Commit

Permalink
remove nullable type handling after sklearn upgrade fixed incompatibi…
Browse files Browse the repository at this point in the history
…lities
  • Loading branch information
Tamar Grey committed Mar 9, 2023
1 parent c6429c2 commit a1fb81c
Showing 1 changed file with 1 addition and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -88,8 +88,7 @@ def transform(self, X, y=None):
pd.DataFrame, pd.Series: Transformed features and target.
"""
X_ww, y_ww = self._prepare_data(X, y)
X_d, y_d = self._handle_nullable_types(X_ww, y_ww)
return super().transform(X_d, y_d)
return super().transform(X_ww, y_ww)

def _get_best_oversampler(self, X):
cat_cols = X.ww.select(
Expand Down

0 comments on commit a1fb81c

Please sign in to comment.