-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
42 lines (37 loc) · 1.33 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
import os
import numpy as np
from setuptools import setup, find_packages
from setuptools.extension import Extension
from Cython.Build import cythonize
extensions = [
Extension('argus_shapes.fast_models', ['argus_shapes/fast_models.pyx'],
include_dirs=[np.get_include()],
extra_compile_args=['-O3']),
Extension('argus_shapes.fast_imgproc', ['argus_shapes/fast_imgproc.pyx'],
include_dirs=[np.get_include()],
extra_compile_args=['-O3'])
]
# Get version and release info, which is all stored in argus_shapes/version.py
ver_file = os.path.join('argus_shapes', 'version.py')
with open(ver_file) as f:
exec(f.read())
opts = dict(name=NAME,
maintainer=MAINTAINER,
maintainer_email=MAINTAINER_EMAIL,
description=DESCRIPTION,
long_description=LONG_DESCRIPTION,
url=URL,
download_url=DOWNLOAD_URL,
license=LICENSE,
classifiers=CLASSIFIERS,
author=AUTHOR,
author_email=AUTHOR_EMAIL,
platforms=PLATFORMS,
version=VERSION,
packages=find_packages(),
package_data=PACKAGE_DATA,
ext_modules=cythonize(extensions),
install_requires=REQUIRES,
requires=REQUIRES)
if __name__ == '__main__':
setup(**opts)