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()
INSTALL_REQUIRES = [
"prototorch>=0.4.4",
"prototorch>=0.5.0",
"pytorch_lightning",
"torchmetrics",
]
CLI = [
"jsonargparse",
]
DEV = [
"bumpversion",
]
@ -43,7 +46,7 @@ TESTS = [
"codecov",
"pytest",
]
ALL = DEV + DOCS + EXAMPLES + TESTS
ALL = CLI + DEV + DOCS + EXAMPLES + TESTS
setup(
name=safe_name("prototorch_" + PLUGIN_NAME),