-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
72 lines (61 loc) · 2.14 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
69
70
71
72
import os
from distutils.core import setup
from setuptools.command.develop import develop
from setuptools.command.install import install
S2V_INSTALL_PATH = '/app/epfml_sent2vec'
class InstallSent2Vec(install):
# Build sent2vec FastText binary, then pip install sent2vec.
def run(self):
os.system("pip3 install -r requirements.txt")
os.system("git clone https://github.com/epfml/sent2vec.git {0}".format(S2V_INSTALL_PATH))
os.system("cd {0} && make".format(S2V_INSTALL_PATH))
os.system("cd {0}/src && python3 setup.py build_ext && pip3 install . ".format(S2V_INSTALL_PATH))
install.run(self)
class DevelopSent2Vec(develop):
# Clone epfml's sent2vec repo, build sent2vec FastText binary, and pip install sent2vec.
def run(self):
os.system("pip3 install -r requirements.txt")
os.system("git clone https://github.com/epfml/sent2vec.git {0}".format(S2V_INSTALL_PATH))
os.system("cd {0} && make".format(S2V_INSTALL_PATH))
os.system("cd {0}/src && python3 setup.py build_ext && pip3 install . ".format(S2V_INSTALL_PATH))
develop.run(self)
current_dir = os.path.dirname(os.path.abspath(__file__))
S2V_INSTALL_PATH = os.getenv('S2V_INSTALL_PATH', os.path.join(current_dir, 'sent2vec'))
'''
setup(
name='nk_sent2vec',
version='1.0.0',
description='Embeds text documents using sent2vec',
author='New Knowledge',
packages=['nk_sent2vec'],
package_data={'': ['*.h, *.cc', '*Makefile']},
include_package_data=True,
install_requires=[
'Cython>=0.28.3',
'numpy>=1.14.1',
'nose>=1.3.7',
],
cmdclass={
'install': InstallSent2Vec,
'develop': DevelopSent2Vec,
}
)
'''
setup(
name='nk_sent2vec',
version='1.1.0',
description='Embeds text documents using sent2vec',
author='New Knowledge',
packages=['nk_sent2vec'],
package_data={'': ['*.h, *.cc', '*Makefile']},
include_package_data=True,
install_requires=[
'Cython>=0.28.3',
'numpy>=1.14.1',
'nose>=1.3.7',
],
cmdclass={
'install': InstallSent2Vec,
'develop': DevelopSent2Vec,
}
)