Unverified Commit 08fad080 authored by Sanchit Gandhi's avatar Sanchit Gandhi Committed by GitHub
Browse files

[modelcard] Check for IterableDataset (#20495)

parent ab9fe452
......@@ -339,9 +339,9 @@ def is_hf_dataset(dataset):
if not is_datasets_available():
return False
from datasets import Dataset
from datasets import Dataset, IterableDataset
return isinstance(dataset, Dataset)
return isinstance(dataset, (Dataset, IterableDataset))
def _get_mapping_values(mapping):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment