-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathsetup.py
executable file
·84 lines (78 loc) · 2.29 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
#! /usr/bin/env python
"""A template."""
import codecs
from setuptools import find_packages, setup
from torch_concepts import __version__
DISTNAME = 'pytorch_concepts'
DESCRIPTION = ' Concept-Based Deep Learning Library for PyTorch.'
with codecs.open('README.md', encoding='utf-8-sig') as f:
LONG_DESCRIPTION = f.read()
MAINTAINER = 'PyC Team'
MAINTAINER_EMAIL = '[email protected]'
URL = 'https://github.com/pyc-team/pytorch_concepts'
LICENSE = 'Apache 2.0'
DOWNLOAD_URL = 'https://github.com/pyc-team/pytorch_concepts'
VERSION = __version__
INSTALL_REQUIRES = [
'numpy',
'opencv-python',
'pandas',
'Pillow==9.5.0',
'scikit-learn',
'scipy',
'torch',
]
CLASSIFIERS = [
'Intended Audience :: Developers',
'Intended Audience :: Science/Research',
'License :: OSI Approved',
'Operating System :: MacOS',
'Operating System :: Microsoft :: Windows',
'Operating System :: POSIX',
'Operating System :: Unix',
'Programming Language :: Python :: 2.7',
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python',
'Topic :: Scientific/Engineering',
'Topic :: Software Development',
]
EXTRAS_REQUIRE = {
'tests': [
'pytest-cov',
'pytest',
],
'docs': [
'matplotlib',
'numpydoc',
'sphinx_rtd_theme',
'sphinx-gallery',
'sphinx',
],
}
setup(
name=DISTNAME,
maintainer=MAINTAINER,
maintainer_email=MAINTAINER_EMAIL,
author=MAINTAINER,
author_email=MAINTAINER_EMAIL,
description=DESCRIPTION,
license=LICENSE,
url=URL,
version=VERSION,
download_url=DOWNLOAD_URL,
long_description=LONG_DESCRIPTION,
long_description_content_type='text/markdown',
zip_safe=False, # the package can run out of an .egg file
classifiers=CLASSIFIERS,
packages=find_packages(include=["torch_concepts", "torch_concepts.*"]),
install_requires=INSTALL_REQUIRES,
extras_require=EXTRAS_REQUIRE,
use_scm_version=True, # Adding so that package data is respected
setup_requires=["setuptools_scm"],
include_package_data=True, # We need this to include static assets (images)
package_data={
"torch_concepts": ["assets/*"],
},
)