def check_regressor_data_not_an_array(name, estimator_orig):
if name in ['TimeSeriesSVR']:
return
X, y = _boston_subset(n_samples=50)
X = pairwise_estimator_convert_X(X, estimator_orig)
y = enforce_estimator_tags_y(estimator_orig, y)
check_estimators_data_not_an_array(name, estimator_orig, X, y)
def check_regressor_data_not_an_array(name, estimator_orig):
if name in ['TimeSeriesSVR']:
return
X, y = _boston_subset(n_samples=50)
X = pairwise_estimator_convert_X(X, estimator_orig)
y = enforce_estimator_tags_y(estimator_orig, y)
for obj_type in ["NotAnArray", "PandasDataframe"]:
if obj_type == "PandasDataframe":
X_ = X[:, :, 0] # pandas df cant be 3d
else:
X_ = X
check_estimators_data_not_an_array(name, estimator_orig, X_, y,
obj_type)