This commit is contained in:
Christian Mantha
2026-03-02 19:10:52 -05:00
commit 2ca0b9ef7c
28907 changed files with 5233713 additions and 0 deletions

View File

@@ -0,0 +1,29 @@
tree_extension_metadata = {
'_tree':
{'sources': ['_tree.pyx'],
'override_options': ['cython_language=cpp', 'optimization=3']},
'_splitter':
{'sources': ['_splitter.pyx'],
'override_options': ['optimization=3']},
'_partitioner':
{'sources': ['_partitioner.pyx'],
'override_options': ['optimization=3']},
'_criterion':
{'sources': ['_criterion.pyx'],
'override_options': ['optimization=3']},
'_utils':
{'sources': ['_utils.pyx'],
'override_options': ['optimization=3']},
}
foreach ext_name, ext_dict : tree_extension_metadata
py.extension_module(
ext_name,
[ext_dict.get('sources'), utils_cython_tree],
dependencies: [np_dep],
override_options : ext_dict.get('override_options', []),
cython_args: cython_args,
subdir: 'sklearn/tree',
install: true
)
endforeach