forked from OpenNMT/OpenNMT-py
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
44 lines (42 loc) · 1.45 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
#!/usr/bin/env python
from setuptools import setup, find_packages
from os import path
this_directory = path.abspath(path.dirname(__file__))
with open(path.join(this_directory, 'README.md'), encoding='utf-8') as f:
long_description = f.read()
setup(
name='OpenNMT-py',
description='A python implementation of OpenNMT',
long_description=long_description,
long_description_content_type='text/markdown',
version='2.1.0',
packages=find_packages(),
project_urls={
"Documentation": "http://opennmt.net/OpenNMT-py/",
"Forum": "http://forum.opennmt.net/",
"Gitter": "https://gitter.im/OpenNMT/OpenNMT-py",
"Source": "https://github.com/OpenNMT/OpenNMT-py/"
},
python_requires=">=3.5",
install_requires=[
"tqdm>=4.51,<5",
"torch==1.6.0",
"torchtext==0.5.0",
"configargparse>=1.2.3,<2",
"tensorboard>=2.3,<3",
"flask==1.1.2",
"waitress==1.4.4",
"pyonmttok>=1.23,<2;platform_system=='Linux' or platform_system=='Darwin'",
"pyyaml==5.4",
],
entry_points={
"console_scripts": [
"onmt_server=onmt.bin.server:main",
"onmt_train=onmt.bin.train:main",
"onmt_translate=onmt.bin.translate:main",
"onmt_release_model=onmt.bin.release_model:main",
"onmt_average_models=onmt.bin.average_models:main",
"onmt_build_vocab=onmt.bin.build_vocab:main"
],
}
)