forked from Tiiiger/bert_score
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
42 lines (40 loc) · 1.47 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
from io import open
from setuptools import find_packages, setup
setup(
name="bert_score",
version='0.3.10',
author="Tianyi Zhang*, Varsha Kishore*, Felix Wu*, Kilian Q. Weinberger, and Yoav Artzi",
author_email="[email protected]",
description="PyTorch implementation of BERT score",
long_description=open("README.md", "r", encoding='utf-8').read(),
long_description_content_type="text/markdown",
keywords='BERT NLP deep learning google metric',
license='MIT',
url="https://github.com/Tiiiger/bert_score",
packages=find_packages(exclude=["*.tests", "*.tests.*",
"tests.*", "tests"]),
install_requires=['torch>=1.0.0',
'pandas>=1.0.1',
'transformers>=3.0.0'
'numpy',
'requests',
'tqdm>=4.31.1',
'matplotlib',
'packaging>=20.9',
],
entry_points={
'console_scripts': [
"bert-score=bert_score_cli.score:main",
"bert-score-show=bert_score_cli.visualize:main",
]
},
include_package_data=True,
python_requires='>=3.6',
tests_require=['pytest'],
classifiers=[
'Intended Audience :: Science/Research',
'License :: OSI Approved :: MIT License',
'Programming Language :: Python :: 3',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
],
)