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

Update installation dependencies

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-04-21 14:51:08 +02:00
parent f4e703abee
commit 3687abc3a7
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
setup.py
View File

@ -20,7 +20,10 @@ DOWNLOAD_URL = "https://github.com/si-cim/prototorch_models.git"
with open("README.md", "r") as fh: with open("README.md", "r") as fh:
long_description = fh.read() long_description = fh.read()
INSTALL_REQUIRES = ["prototorch"] INSTALL_REQUIRES = ["prototorch", "pytorch_lightning"]
EXAMPLES = ["matplotlib", "scikit-learn"]
TESTS = ["pytest"]
ALL = EXAMPLES + TESTS
setup( setup(
name=safe_name("prototorch_" + PLUGIN_NAME), name=safe_name("prototorch_" + PLUGIN_NAME),
@ -34,7 +37,11 @@ setup(
license="MIT", license="MIT",
install_requires=INSTALL_REQUIRES, install_requires=INSTALL_REQUIRES,
setup_requires=["setuptools_scm"], setup_requires=["setuptools_scm"],
extras_require={}, extras_require={
"examples": EXAMPLES,
"tests": TESTS,
"all": ALL,
},
classifiers=[ classifiers=[
"Development Status :: 2 - Pre-Alpha", "Development Status :: 2 - Pre-Alpha",
"Environment :: Plugins", "Environment :: Plugins",
@ -46,12 +53,15 @@ setup(
"Programming Language :: Python :: 3.6", "Programming Language :: Python :: 3.6",
"Programming Language :: Python :: 3.7", "Programming Language :: Python :: 3.7",
"Programming Language :: Python :: 3.8", "Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Operating System :: OS Independent", "Operating System :: OS Independent",
"Topic :: Scientific/Engineering :: Artificial Intelligence", "Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Software Development :: Libraries", "Topic :: Software Development :: Libraries",
"Topic :: Software Development :: Libraries :: Python Modules", "Topic :: Software Development :: Libraries :: Python Modules",
], ],
entry_points={"prototorch.plugins": f"{PLUGIN_NAME} = prototorch.{PLUGIN_NAME}"}, entry_points={
"prototorch.plugins": f"{PLUGIN_NAME} = prototorch.{PLUGIN_NAME}"
},
packages=find_namespace_packages(include=["prototorch.*"]), packages=find_namespace_packages(include=["prototorch.*"]),
zip_safe=False, zip_safe=False,
) )