scikit-learn · thomasjpfan · Aug 1, 2023 · Mar 17, 2023 · Mar 20, 2023 · Mar 21, 2023
diff --git a/.gitignore b/.gitignore
@@ -99,6 +99,9 @@ sklearn/metrics/_pairwise_distances_reduction/_middle_term_computer.pxd
 sklearn/metrics/_pairwise_distances_reduction/_middle_term_computer.pyx
 sklearn/metrics/_pairwise_distances_reduction/_radius_neighbors.pxd
 sklearn/metrics/_pairwise_distances_reduction/_radius_neighbors.pyx
+sklearn/neighbors/_ball_tree.pyx
+sklearn/neighbors/_binary_tree.pxi
+sklearn/neighbors/_kd_tree.pyx
 
 # Default JupyterLite content
 jupyterlite_contents
diff --git a/setup.cfg b/setup.cfg
@@ -53,6 +53,9 @@ ignore =
     sklearn/metrics/_pairwise_distances_reduction/_middle_term_computer.pyx
     sklearn/metrics/_pairwise_distances_reduction/_radius_neighbors.pxd
     sklearn/metrics/_pairwise_distances_reduction/_radius_neighbors.pyx
+    sklearn/neighbors/_ball_tree.pyx
+    sklearn/neighbors/_binary_tree.pxi
+    sklearn/neighbors/_kd_tree.pyx
 
 
 [codespell]

diff --git a/setup.py b/setup.py
@@ -306,8 +306,9 @@ def check_package_status(package, min_version):
         },
     ],
     "neighbors": [
-        {"sources": ["_ball_tree.pyx"], "include_np": True},
-        {"sources": ["_kd_tree.pyx"], "include_np": True},
+        {"sources": ["_binary_tree.pxi.tp"], "include_np": True},
+        {"sources": ["_ball_tree.pyx.tp"], "include_np": True},
+        {"sources": ["_kd_tree.pyx.tp"], "include_np": True},
         {"sources": ["_partition_nodes.pyx"], "language": "c++", "include_np": True},
         {"sources": ["_quad_tree.pyx"], "include_np": True},
     ],
@@ -499,13 +500,18 @@ def configure_extension_modules():
                 # `source` is a Tempita file
                 tempita_sources.append(source)
 
-                # Do not include pxd files that were generated by tempita
-                if os.path.splitext(new_source_path)[-1] == ".pxd":
-                    continue
-                sources.append(new_source_path)
+                # Only include source files that are pyx files
+                if os.path.splitext(new_source_path)[-1] == ".pyx":
+                    sources.append(new_source_path)
 
             gen_from_templates(tempita_sources)
 
+            # Do not progress if we only have a tempita file which we don't
+            # want to include like the .pxi.tp extension. In such a case
+            # sources would be empty.
+            if not sources:
+                continue
+
             # By convention, our extensions always use the name of the first source
             source_name = os.path.splitext(os.path.basename(sources[0]))[0]
             if submodule:

diff --git a/sklearn/neighbors/_ball_tree.pyx b/sklearn/neighbors/_ball_tree.pyx