This repository has been archived by the owner on Aug 31, 2023. It is now read-only.
forked from QData/TextAttack
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
75 lines (68 loc) · 1.95 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
# Version number is tracked in docs/conf.py.
import setuptools
from docs import conf as docs_conf
with open("README.md", "r", encoding="utf8") as fh:
long_description = fh.read()
extras = {}
# Packages required for installing docs.
extras["docs"] = ["recommonmark", "nbsphinx", "sphinx-autobuild", "sphinx-rtd-theme"]
# Packages required for formatting code & running tests.
extras["test"] = [
"black==20.8b1",
"docformatter",
"isort==5.6.4",
"flake8",
"pytest",
"pytest-xdist",
]
extras["tensorflow"] = [
"tensorflow>=2",
"tensorflow_hub",
"tensorflow_text>=2",
"tensorboardX",
]
extras["optional"] = [
"sentence_transformers>0.2.6",
"stanza",
"visdom",
"wandb",
"gensim==3.8.3",
]
# For developers, install development tools along with all optional dependencies.
extras["dev"] = (
extras["docs"] + extras["test"] + extras["tensorflow"] + extras["optional"]
)
setuptools.setup(
name="textattack",
version=docs_conf.release,
author="QData Lab at the University of Virginia",
author_email="[email protected]",
description="A library for generating text adversarial examples",
include_package_data=False,
license="MIT",
long_description=long_description,
long_description_content_type="text/markdown",
url="https://github.com/QData/textattack",
packages=setuptools.find_namespace_packages(
exclude=[
"build*",
"docs*",
"dist*",
"examples*",
"outputs*",
"tests*",
"wandb*",
]
),
extras_require=extras,
entry_points={
"console_scripts": ["textattack=textattack.commands.textattack_cli:main"],
},
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
],
python_requires=">=3.6",
install_requires=open("requirements.txt").readlines(),
)