Skip to content

Commit dd03529

Browse files
committed
fix wrong return type
1 parent 290734f commit dd03529

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

libmultilabel/nn/data_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,7 @@ def _load_raw_data(data, is_test=False, tokenize_text=True, remove_no_label_data
159159
This is effective only when is_test=False. Defaults to False.
160160
161161
Returns:
162-
pandas.DataFrame: Data composed of index, label, and tokenized text.
162+
dict: [{(optional: "index": ..., )"label": ..., "text": ...}, ...]
163163
"""
164164
assert isinstance(data, str) or isinstance(data, pd.DataFrame), "Data must be from a file or pandas dataframe."
165165
if isinstance(data, str):

0 commit comments

Comments
 (0)