Skip to content

Commit 6c0a05b

Browse files
author
Felipe Mello
committed
add missing doc
1 parent 09c2619 commit 6c0a05b

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

torchtune/datasets/_text_completion.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,9 @@ class TextCompletionDataset(Dataset):
3333
filter_fn (Optional[Callable]): callable used to filter the dataset prior to any pre-processing. See
3434
the Hugging Face `docs <https://huggingface.co/docs/datasets/v2.20.0/process#select-and-filter>`_ for more
3535
details.
36-
**load_dataset_kwargs (Dict[str, Any]): additional keyword arguments to pass to ``load_dataset``,
37-
such as ``data_files`` or ``split``.
36+
**load_dataset_kwargs (Dict[str, Any]): additional keyword arguments to pass to ``load_dataset``. See Hugging
37+
Face's `API ref <https://huggingface.co/docs/datasets/en/package_reference/loading_methods#datasets.load_dataset>`_
38+
for more details.
3839
"""
3940

4041
def __init__(

0 commit comments

Comments
 (0)