diff --git a/src/datasets/formatting/dataset_wrappers/torch_iterable_dataset.py b/src/datasets/formatting/dataset_wrappers/torch_iterable_dataset.py index 33ce181aaad..78b0803f2f7 100644 --- a/src/datasets/formatting/dataset_wrappers/torch_iterable_dataset.py +++ b/src/datasets/formatting/dataset_wrappers/torch_iterable_dataset.py @@ -1,4 +1,4 @@ -import fsspec +import fsspec.asyn import torch from ...iterable_dataset import IterableDataset, _apply_feature_types