-
Notifications
You must be signed in to change notification settings - Fork 4
Expand file tree
/
Copy pathsetup.py
More file actions
31 lines (28 loc) · 996 Bytes
/
Copy pathsetup.py
File metadata and controls
31 lines (28 loc) · 996 Bytes
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
from setuptools import setup, find_packages
from setuptools.extension import Extension
from Cython.Build import cythonize
import numpy
extensions = [
Extension(
"woodtapper.extract_rules.Splitter.QuantileSplitter", # Full module path
[
"woodtapper/extract_rules/Splitter/_QuantileSplitter.pyx"
], # Path to .pyx file
),
Extension(
"woodtapper.example_sampling.utils.weights", # Full module path
["woodtapper/example_sampling/utils/weights.pyx"], # Path to .pyx file
),
]
with open("README.md", encoding="utf-8") as f:
long_description = f.read()
setup(
name="woodtapper", # must be unique on PyPI
author="Abdoulaye SAKHO",
author_email="abdoulaye.sakho@artefact.com",
description="A Python toolbox for interpretable and explainable tree ensembles.",
long_description=long_description,
packages=find_packages(),
ext_modules=cythonize(extensions),
include_dirs=[numpy.get_include()],
)