-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
38 lines (31 loc) · 1.21 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
from setuptools import setup, find_packages
__version__ = "1.0"
# add readme
with open('README.md', 'r') as f:
LONG_DESCRIPTION = f.read()
# add dependencies
with open('requirements.txt', 'r') as f:
INSTALL_REQUIRES = f.read().strip().split('\n')
setup(
name='AdversarialTensors',
version=__version__,
author='Manish Bhattarai, Mehmet Cagri, Ben Nebgen, Kim Rasmussen, Boian Alexandrov',
author_email='[email protected]',
long_description=LONG_DESCRIPTION,
url='https://github.com/lanl/AdversarialTensors', # change this to GitHub once published
description='AdversarialTensors: Tensors-based framework for adversarial robustness',
setup_requires=['numpy', 'scipy', 'matplotlib', 'torch', 'torchvision', 'tensorly','scikit-learn', 'pytest'],
install_requires=INSTALL_REQUIRES,
packages=find_packages(),
python_requires='>=3.9',
classifiers=[
'Development Status :: ' + str(__version__) + ' - Beta',
'Programming Language :: Python :: 3.9',
'Topic :: Machine Learning :: Libraries'
],
license='License :: BSD3 License',
zip_safe=False,
dependency_links=[
'git+https://github.com/fra31/auto-attack.git#egg=auto-attack'
]
)