Skip to content

Commit

Permalink
Fix datasets download imports (#143)
Browse files Browse the repository at this point in the history
* Fix datasets download imports

* Fix dataset download imports in inspect

Co-authored-by: leandro <[email protected]>
  • Loading branch information
albertvillanova and leandro authored Jun 15, 2022
1 parent d638dba commit 3fb6e68
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 4 deletions.
2 changes: 1 addition & 1 deletion src/evaluate/inspect.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
from typing import Optional

import requests
from datasets.utils import DownloadConfig
from datasets import DownloadConfig

from .config import EVALUATION_MODULE_TYPES, HF_LIST_ENDPOINT
from .loading import evaluation_module_factory
Expand Down
2 changes: 1 addition & 1 deletion src/evaluate/loading.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,9 +27,9 @@
from typing import List, Optional, Tuple, Type, Union
from urllib.parse import urlparse

from datasets import DownloadMode
from datasets.builder import DatasetBuilder
from datasets.packaged_modules import _EXTENSION_TO_MODULE, _hash_python_lines
from datasets.utils.download_manager import DownloadMode
from datasets.utils.filelock import FileLock
from datasets.utils.version import Version

Expand Down
3 changes: 1 addition & 2 deletions src/evaluate/module.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,13 +21,12 @@

import numpy as np
import pyarrow as pa
from datasets import DatasetInfo
from datasets import DatasetInfo, DownloadManager
from datasets.arrow_dataset import Dataset
from datasets.arrow_reader import ArrowReader
from datasets.arrow_writer import ArrowWriter
from datasets.features import Features, Sequence, Value
from datasets.features.features import _check_non_null_non_empty_recursive
from datasets.utils.download_manager import DownloadManager
from datasets.utils.filelock import BaseFileLock, FileLock, Timeout
from datasets.utils.py_utils import copyfunc, temp_seed, zip_dict

Expand Down

0 comments on commit 3fb6e68

Please sign in to comment.