scikit-learn
diff --git a/‎setup.py
Copy file name to clipboardExpand all lines: setup.py
+9-6Lines changed: 9 additions & 6 deletions b/‎setup.py
Copy file name to clipboardExpand all lines: setup.py
+9-6Lines changed: 9 additions & 6 deletions
@@ -79,17 +79,20 @@ def run(self):
             shutil.rmtree("build")
         for dirpath, dirnames, filenames in os.walk("sklearn"):
             for filename in filenames:
-                if any(
-                    filename.endswith(suffix)
-                    for suffix in (".so", ".pyd", ".dll", ".pyc")
-                ):
+                root, extension = os.path.splitext(filename)
+
+                if extension in [".so", ".pyd", ".dll", ".pyc"]:
                     os.unlink(os.path.join(dirpath, filename))
-                    continue
-                extension = os.path.splitext(filename)[1]
+
                 if remove_c_files and extension in [".c", ".cpp"]:
                     pyx_file = str.replace(filename, extension, ".pyx")
                     if os.path.exists(os.path.join(dirpath, pyx_file)):
                         os.unlink(os.path.join(dirpath, filename))
+
+                if remove_c_files and extension == ".tp":
+                    if os.path.exists(os.path.join(dirpath, root)):
+                        os.unlink(os.path.join(dirpath, root))
+
             for dirname in dirnames:
                 if dirname == "__pycache__":
                     shutil.rmtree(os.path.join(dirpath, dirname))