|
@@ -264,8 +264,8 @@ class PipelineSelector(ABC):
|
|
|
|
|
|
else:
|
|
|
try:
|
|
|
- assert(isinstance(X_train, (pd.DataFrame, np.array)) and
|
|
|
- isinstance(y_train, (pd.Series, np.array,
|
|
|
+ assert(isinstance(X_train, (pd.DataFrame, np.ndarray)) and
|
|
|
+ isinstance(y_train, (pd.Series, np.ndarray,
|
|
|
pd.DataFrame, NoneType)) and
|
|
|
(X_val is None) and (y_val is None))
|
|
|
except AssertionError:
|