-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
56 lines (51 loc) · 1.51 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
from distutils import sysconfig
from setuptools import setup, Extension
import os
cpp_path = './src/lib'
cpp_args = ['-std=c++11']
ext_modules = [
Extension(
'hdp_funcs',
[os.path.join(cpp_path,'hdp_funcs.cpp')],
include_dirs=[os.path.join(cpp_path,'pybind11/include'),
os.path.join(cpp_path,'eigen')],
language='c++',
extra_compile_args = cpp_args,
),
Extension(
'hdp_preproc',
[os.path.join(cpp_path,'hdp_preproc.cpp')],
include_dirs=[os.path.join(cpp_path,'pybind11/include'),
os.path.join(cpp_path,'eigen')],
language='c++',
extra_compile_args = cpp_args,
),
]
setup(name='hdp',
version='0.1.0',
description='Implementation of Hierarchical LDA',
url='http://github.com/datadiarist/hplda',
author='Andrew and Eduardo',
author_email='andrew.carr@duke.edu',
license='MIT',
classifiers=[
'Programming Language :: Python :: 3',
'License :: OSI Approved :: MIT License',
'Topic :: Text Processing :: Linguistic',
'Operating System :: OS Independent'
],
keywords='topic non-parametric modeling nlp',
python_requires = '>=3.0',
package_dir = {'': 'src'},
packages=['hdp'],
zip_safe=False,
install_requires=[
'pybind11',
'pandas',
'scipy',
'numpy',
'nltk'],
ext_modules=ext_modules,
test_suite='nose.collector',
tests_require=['nose']
)