-
Notifications
You must be signed in to change notification settings - Fork 9
/
setup.py
68 lines (59 loc) · 2 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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""The setup script."""
from setuptools import setup, find_packages
from torch.utils.cpp_extension import CppExtension, BuildExtension
from os.path import abspath, dirname
def load_requirements(path):
with open(path) as fin:
return [
line
for line in map(lambda l: l.strip(), fin.readlines())
if line and not line.startswith('#')
]
with open('README.rst') as readme_file:
readme = readme_file.read()
with open('HISTORY.rst') as history_file:
history = history_file.read()
requirements = load_requirements('requirements_prod.txt')
test_requirements = load_requirements('requirements_dev.txt')
setup(
author="Hunt Zhan",
author_email='huntzhan.dev@gmail.com',
classifiers=[
'Development Status :: 2 - Pre-Alpha',
'Intended Audience :: Developers',
'License :: OSI Approved :: MIT License',
'Natural Language :: English',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
],
description="None",
install_requires=requirements,
license="MIT license",
long_description=readme + '\n\n' + history,
include_package_data=True,
keywords='pytorch_fast_elmo',
name='pytorch_fast_elmo',
packages=find_packages(include=['pytorch_fast_elmo']),
test_suite='tests',
tests_require=test_requirements,
url='https://github.com/cnt-dev/pytorch-fast-elmo',
version='0.3.2',
zip_safe=False,
# Pytorch Cpp Extension.
ext_modules=[
CppExtension(
'_pytorch_fast_elmo',
[
'extension/elmo_character_encoder.cc',
'extension/scalar_mix.cc',
'extension/bind.cc',
],
include_dirs=[dirname(abspath(__file__))],
),
],
cmdclass={'build_ext': BuildExtension},
entry_points={'console_scripts': ['fast-elmo = pytorch_fast_elmo.cli:main']},
)