diff --git a/ml_datasets/loaders/mnist.py b/ml_datasets/loaders/mnist.py index 1575726..2e9c803 100644 --- a/ml_datasets/loaders/mnist.py +++ b/ml_datasets/loaders/mnist.py @@ -1,8 +1,9 @@ import random import zipfile import gzip + +import cloudpickle as pickle import numpy as np -from srsly import cloudpickle as pickle from ..util import unzip, to_categorical, get_file from .._registry import register_loader diff --git a/ml_datasets/loaders/reuters.py b/ml_datasets/loaders/reuters.py index 7a9af4c..367abfc 100644 --- a/ml_datasets/loaders/reuters.py +++ b/ml_datasets/loaders/reuters.py @@ -1,4 +1,4 @@ -from srsly import cloudpickle as pickle +import cloudpickle as pickle import numpy from ..util import get_file diff --git a/requirements.txt b/requirements.txt index 5773b45..0b76345 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,7 @@ +cloudpickle>=2.2 numpy>=1.7.0 scipy>=1.7.0 tqdm>=4.10.0,<5.0.0 # Our libraries -srsly>=1.0.1,<3.0.0 +srsly>=1.0.1,<4.0.0 catalogue>=0.2.0,<3.0.0 diff --git a/setup.cfg b/setup.cfg index 512be93..9d40de5 100644 --- a/setup.cfg +++ b/setup.cfg @@ -1,5 +1,5 @@ [metadata] -version = 0.2.0 +version = 0.2.1 description = Machine Learning dataset loaders url = https://github.com/explosion/ml-datasets author = Explosion @@ -13,10 +13,11 @@ zip_safe = true include_package_data = true python_requires = >=3.6 install_requires = + cloudpickle>=2.2 numpy>=1.7.0 tqdm>=4.10.0,<5.0.0 # Our libraries - srsly>=1.0.1,<3.0.0 + srsly>=1.0.1,<4.0.0 catalogue>=0.2.0,<3.0.0 [options.entry_points]