diff --git a/src/lighteval/logging/evaluation_tracker.py b/src/lighteval/logging/evaluation_tracker.py index af5784bc2..585d74232 100644 --- a/src/lighteval/logging/evaluation_tracker.py +++ b/src/lighteval/logging/evaluation_tracker.py @@ -34,7 +34,6 @@ import torch from datasets import Dataset, load_dataset from datasets.utils.metadata import MetadataConfigs -from fsspec import url_to_fs from huggingface_hub import DatasetCard, DatasetCardData, HfApi, HFSummaryWriter, hf_hub_url from lighteval.logging.info_loggers import ( @@ -53,6 +52,11 @@ if is_nanotron_available(): from nanotron.config import GeneralArgs # type: ignore +try: + from fsspec import url_to_fs +except ImportError: + from fsspec.core import url_to_fs + class EnhancedJSONEncoder(json.JSONEncoder): """