Support for srsly 3.0 (#6)

Author: crusaderky

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

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

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

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]