Fix include .pxd definitions for sklearn.tree in wheels (#12381)

scikit-learn · Nov 14, 2018 · 23f06ca · 23f06ca
1 parent ecad9b4
commit 23f06ca
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sklearn/tree/setup.py b/sklearn/tree/setup.py
@@ -31,6 +31,10 @@ def configuration(parent_package="", top_path=None):
                          extra_compile_args=["-O3"])
 
     config.add_subpackage("tests")
+    config.add_data_files("_criterion.pxd")
+    config.add_data_files("_splitter.pxd")
+    config.add_data_files("_tree.pxd")
+    config.add_data_files("_utils.pxd")
 
     return config