mirror of
https://github.com/QData/TextAttack.git
synced 2021-10-13 00:05:06 +03:00
82 lines
2.0 KiB
Python
82 lines
2.0 KiB
Python
# 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",
|
|
"sphinx-markdown-tables",
|
|
]
|
|
# 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="jm8wx@virginia.edu",
|
|
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(),
|
|
)
|