-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
49 lines (46 loc) · 1.49 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
from setuptools import setup, find_packages
import codecs
import os.path
def read(rel_path):
here = os.path.abspath(os.path.dirname(__file__))
with codecs.open(os.path.join(here, rel_path), 'r') as fp:
return fp.read()
def get_version(rel_path):
for line in read(rel_path).splitlines():
if line.startswith('__version__'):
delim = '"' if '"' in line else "'"
return line.split(delim)[1]
else:
raise RuntimeError("Unable to find version string.")
setup(
name='argminer',
version=get_version("argminer/__init__.py"), #'0.0.12',
description='A package for processing SOTA argument mining datasets',
author='Yousef Nami',
author_email='[email protected]',
url='https://github.com/namiyousef/argument-mining',
install_requires=[
'torch',
'transformers',
'sentencepiece',
'pandas',
'plac',
'numpy',
'tqdm',
'sklearn',
'protobuf',
'colab-dev-tools',
'matplotlib',
'ml-dev-tools',
'connexion[swagger-ui]'
#'pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html'
],
#package_data={}
packages=find_packages(exclude=('tests*', 'experiments*')),
package_data={'': ['api/specs/api.yaml']},
include_package_data=True,
license='MIT',
entry_points={
'console_scripts': ['argminer-api=argminer.run_api:main'],
}
)