Skip to content

Commit cefa2aa

Browse files
committed
Update _sklearnadapter.py
1 parent 672b40c commit cefa2aa

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

src/hyperactive/integrations/sklearn/_adapter/_sklearnadapter.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,12 +18,14 @@ def _refit(self, X, y=None, **fit_params):
1818

1919
def _check_data(self, X, y):
2020
X, y = indexable(X, y)
21+
if hasattr(X, "ndim") and X.ndim == 1:
22+
X = X.reshape(-1, 1)
2123
if hasattr(self, "_validate_data"):
2224
validate_data = self._validate_data
2325
else:
2426
from sklearn.utils.validation import validate_data
2527

26-
return validate_data(X, y, ensure_2d=False)
28+
return validate_data(X, y)
2729

2830
@property
2931
def fit_successful(self):

0 commit comments

Comments
 (0)