prototorch_models/setup.py

70 lines
2.4 KiB
Python
Raw Normal View History

2021-04-21 11:13:28 +00:00
"""
2021-04-21 11:30:50 +00:00
_____ _ _______ _
| __ \ | | |__ __| | |
2021-04-21 11:13:28 +00:00
| |__) | __ ___ | |_ ___ | | ___ _ __ ___| |__
2021-04-21 11:30:50 +00:00
| ___/ '__/ _ \| __/ _ \| |/ _ \| '__/ __| '_ \
2021-04-21 11:13:28 +00:00
| | | | | (_) | || (_) | | (_) | | | (__| | | |
|_| |_| \___/ \__\___/|_|\___/|_| \___|_| |_|Plugin
2021-04-21 11:30:50 +00:00
ProtoTorch models Plugin Package
2021-04-21 11:13:28 +00:00
"""
from pkg_resources import safe_name
2021-04-23 15:27:47 +00:00
from setuptools import find_namespace_packages, setup
2021-04-21 11:13:28 +00:00
2021-04-21 11:30:50 +00:00
PLUGIN_NAME = "models"
2021-04-21 11:13:28 +00:00
2021-04-21 11:30:50 +00:00
PROJECT_URL = "https://github.com/si-cim/prototorch_models"
DOWNLOAD_URL = "https://github.com/si-cim/prototorch_models.git"
2021-04-21 11:13:28 +00:00
with open("README.md", "r") as fh:
long_description = fh.read()
INSTALL_REQUIRES = ["prototorch>=0.4.1", "pytorch_lightning", "torchmetrics"]
2021-05-10 13:50:06 +00:00
DEV = ["bumpversion"]
2021-04-21 12:51:08 +00:00
EXAMPLES = ["matplotlib", "scikit-learn"]
2021-05-10 13:32:52 +00:00
TESTS = ["codecov", "pytest"]
2021-05-10 13:50:06 +00:00
ALL = DEV + EXAMPLES + TESTS
2021-04-21 11:13:28 +00:00
setup(
name=safe_name("prototorch_" + PLUGIN_NAME),
2021-05-11 11:41:26 +00:00
version="0.1.6",
2021-05-10 13:50:06 +00:00
description="Pre-packaged prototype-based "
"machine learning models using ProtoTorch and PyTorch-Lightning.",
2021-04-21 11:13:28 +00:00
long_description=long_description,
2021-05-10 15:12:54 +00:00
long_description_content_type="text/markdown",
2021-04-21 11:13:28 +00:00
author="Alexander Engelsberger",
author_email="engelsbe@hs-mittweida.de",
url=PROJECT_URL,
download_url=DOWNLOAD_URL,
license="MIT",
install_requires=INSTALL_REQUIRES,
2021-04-21 12:51:08 +00:00
extras_require={
2021-05-10 13:50:06 +00:00
"dev": DEV,
2021-04-21 12:51:08 +00:00
"examples": EXAMPLES,
"tests": TESTS,
"all": ALL,
},
2021-04-21 11:13:28 +00:00
classifiers=[
"Development Status :: 2 - Pre-Alpha",
"Environment :: Plugins",
"Intended Audience :: Developers",
"Intended Audience :: Education",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Natural Language :: English",
"Programming Language :: Python :: 3.6",
"Programming Language :: Python :: 3.7",
"Programming Language :: Python :: 3.8",
2021-04-21 12:51:08 +00:00
"Programming Language :: Python :: 3.9",
2021-04-21 11:13:28 +00:00
"Operating System :: OS Independent",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Software Development :: Libraries",
"Topic :: Software Development :: Libraries :: Python Modules",
],
2021-04-21 12:51:08 +00:00
entry_points={
"prototorch.plugins": f"{PLUGIN_NAME} = prototorch.{PLUGIN_NAME}"
},
2021-04-21 11:13:28 +00:00
packages=find_namespace_packages(include=["prototorch.*"]),
zip_safe=False,
)