Skip to content

Commit

Permalink
adding in import to fix doctests
Browse files Browse the repository at this point in the history
  • Loading branch information
omatthew98 committed Jan 25, 2024
1 parent edd4c16 commit 65023d6
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions python/ray/data/datasource/huggingface_datasource.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,6 +79,8 @@ def list_parquet_urls_from_dataset(
"""Return list of Hugging Face hosted parquet file URLs if they
exist for the data (i.e. if the dataset is a public dataset that
has not been transformed) else return an empty list."""
import datasets

# We can use the dataset name, config name, and split name to load
# public hugging face datasets from the Hugging Face Hub. More info
# here: https://huggingface.co/docs/datasets-server/parquet
Expand Down

0 comments on commit 65023d6

Please sign in to comment.