-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathsetup.py
28 lines (23 loc) · 837 Bytes
/
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
from setuptools import setup, find_packages
from distutils.extension import Extension
from Cython.Build import cythonize
import numpy
import os
def read_file(filename):
with open(os.path.join(os.path.dirname(__file__), filename)) as file:
return file.read()
extensions = [
Extension("sklearn_quantile.utils.weighted_quantile",
["sklearn_quantile/utils/weighted_quantile.pyx"]),
Extension("sklearn_quantile.ensemble.quantile",
["sklearn_quantile/ensemble/quantile.pyx"]),
Extension("sklearn_quantile.ensemble.maximum",
["sklearn_quantile/ensemble/maximum.pyx"])
]
setup(
name="sklearn-quantile",
version="0.0.29",
packages=find_packages(),
ext_modules=cythonize(extensions, annotate=True, language_level="3"),
include_dirs=[numpy.get_include()]
)