Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion ml_datasets/loaders/mnist.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down
2 changes: 1 addition & 1 deletion ml_datasets/loaders/reuters.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from srsly import cloudpickle as pickle
import cloudpickle as pickle
import numpy

from ..util import get_file
Expand Down
3 changes: 2 additions & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -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
5 changes: 3 additions & 2 deletions setup.cfg
Original file line number Diff line number Diff line change
@@ -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
Expand All @@ -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]
Expand Down