Update dependencies
This commit is contained in:
parent
ee42fd68b1
commit
30adbf705c
3
setup.py
3
setup.py
@ -20,10 +20,10 @@ INSTALL_REQUIRES = [
|
|||||||
"torch>=1.3.1",
|
"torch>=1.3.1",
|
||||||
"torchvision>=0.5.0",
|
"torchvision>=0.5.0",
|
||||||
"numpy>=1.9.1",
|
"numpy>=1.9.1",
|
||||||
|
"sklearn",
|
||||||
]
|
]
|
||||||
DATASETS = [
|
DATASETS = [
|
||||||
"requests",
|
"requests",
|
||||||
"sklearn",
|
|
||||||
"tqdm",
|
"tqdm",
|
||||||
]
|
]
|
||||||
DEV = ["bumpversion"]
|
DEV = ["bumpversion"]
|
||||||
@ -35,7 +35,6 @@ DOCS = [
|
|||||||
"sphinx-autodoc-typehints",
|
"sphinx-autodoc-typehints",
|
||||||
]
|
]
|
||||||
EXAMPLES = [
|
EXAMPLES = [
|
||||||
"sklearn",
|
|
||||||
"matplotlib",
|
"matplotlib",
|
||||||
"torchinfo",
|
"torchinfo",
|
||||||
]
|
]
|
||||||
|
Loading…
Reference in New Issue
Block a user