-
Notifications
You must be signed in to change notification settings - Fork 21
/
setup.py
49 lines (45 loc) · 1.42 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, Extension
try:
import numpy as np
except ImportError:
exit('Please install numpy first.\nUse pip install numpy.')
try:
from Cython.Build import cythonize
from Cython.Distutils import build_ext
except ImportError:
exit('You need Cython too :(.\n Use pip install cython.\nNo more requirements, promise!')
extensions = [
Extension(
'reco.recommender.funksvd',
['reco/recommender/funksvd.pyx'],
include_dirs=[np.get_include()]
),
Extension(
'reco.recommender.fm',
['reco/recommender/fm.pyx'],
include_dirs=[np.get_include()]
)
]
ext_modules = cythonize(extensions)
setup(name='reco',
version='0.2.1',
description='a simple yet powerful recommender systems library in python',
url='http://github.com/mayukh18/reco',
author='Mayukh Bhattacharyya',
author_email='[email protected]',
license='MIT',
download_url = 'https://github.com/mayukh18/reco/tarball/0.2.1',
include_package_data = True,
keywords=['recommendation'],
packages=['reco',
'reco.recommender',
'reco.datasets',
'reco.metrics',
'reco.cross_validation'],
ext_modules = ext_modules,
cmdclass= {'build_ext': build_ext},
install_requires=['numpy',
'pandas',
],
zip_safe=False
)