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

Merge pull request #6 from si-cim/main

Browse Source 
Pull changes to dev from main
This commit is contained in:
Jensun Ravichandran 2021-05-10 15:47:55 +02:00 committed by GitHub
commit 95c21dd9ca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
setup.py
View File

@ -20,15 +20,16 @@ with open("README.md", "r") as fh:
long_description = fh.read() long_description = fh.read()
INSTALL_REQUIRES = ["prototorch", "pytorch_lightning", "torchmetrics"] INSTALL_REQUIRES = ["prototorch", "pytorch_lightning", "torchmetrics"]
DEV = ["bumpversion"]
EXAMPLES = ["matplotlib", "scikit-learn"] EXAMPLES = ["matplotlib", "scikit-learn"]
TESTS = ["codecov", "pytest"] TESTS = ["codecov", "pytest"]
ALL = EXAMPLES + TESTS ALL = DEV + EXAMPLES + TESTS
setup( setup(
name=safe_name("prototorch_" + PLUGIN_NAME), name=safe_name("prototorch_" + PLUGIN_NAME),
version="0.0.0", version="0.0.0",
descripion= description="Pre-packaged prototype-based "
"Pre-packaged prototype-based machine learning models using ProtoTorch and PyTorch-Lightning.", "machine learning models using ProtoTorch and PyTorch-Lightning.",
long_description=long_description, long_description=long_description,
author="Alexander Engelsberger", author="Alexander Engelsberger",
author_email="engelsbe@hs-mittweida.de", author_email="engelsbe@hs-mittweida.de",
@ -37,6 +38,7 @@ setup(
license="MIT", license="MIT",
install_requires=INSTALL_REQUIRES, install_requires=INSTALL_REQUIRES,
extras_require={ extras_require={
"dev": DEV,
"examples": EXAMPLES, "examples": EXAMPLES,
"tests": TESTS, "tests": TESTS,
"all": ALL, "all": ALL,