julius/prototorch_models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update setup.py

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-06-02 13:01:27 +02:00
parent ef6bcc1079
commit 86688b26b0
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
setup.py
View File

@ -20,10 +20,13 @@ with open("README.md", "r") as fh:
long_description = fh.read() long_description = fh.read()
INSTALL_REQUIRES = [ INSTALL_REQUIRES = [
"prototorch>=0.4.4", "prototorch>=0.5.0",
"pytorch_lightning", "pytorch_lightning",
"torchmetrics", "torchmetrics",
] ]
CLI = [
"jsonargparse",
]
DEV = [ DEV = [
"bumpversion", "bumpversion",
] ]
@ -43,7 +46,7 @@ TESTS = [
"codecov", "codecov",
"pytest", "pytest",
] ]
ALL = DEV + DOCS + EXAMPLES + TESTS ALL = CLI + DEV + DOCS + EXAMPLES + TESTS
setup( setup(
name=safe_name("prototorch_" + PLUGIN_NAME), name=safe_name("prototorch_" + PLUGIN_NAME),