Add condition for pandas
@@ -153,6 +153,7 @@ def _daal_train_test_split(*arrays, **options):
else:
raise ValueError('Array can\'t be converted to needed format')
+ if pandas_is_imported:
if isinstance(arr, pd.core.frame.DataFrame):
train_arr, test_arr = pd.DataFrame(train_arr), pd.DataFrame(test_arr)
if isinstance(arr, pd.core.series.Series):
