Skip to content

Commit 29a023c

Browse files
amuellermfeurer
authored andcommitted
don't warn if we can convert to dataframe (#829)
* don't warn if we can convert to dataframe * don't convert
1 parent 23d4e6f commit 29a023c

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

openml/datasets/dataset.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -531,8 +531,11 @@ def _encode_if_category(column):
531531
'PyOpenML cannot handle string when returning numpy'
532532
' arrays. Use dataset_format="dataframe".'
533533
)
534-
elif array_format == "dataframe" and scipy.sparse.issparse(data):
535-
return pd.SparseDataFrame(data, columns=attribute_names)
534+
elif array_format == "dataframe":
535+
if scipy.sparse.issparse(data):
536+
return pd.SparseDataFrame(data, columns=attribute_names)
537+
else:
538+
return data
536539
else:
537540
data_type = "sparse-data" if scipy.sparse.issparse(data) else "non-sparse data"
538541
logging.warning(

0 commit comments

Comments
 (0)