fix(//py): Add new dirs to remove during clean

Signed-off-by: Naren Dasan <naren@narendasan.com> Signed-off-by: Naren Dasan <narens@nvidia.com>
pytorch · Oct 20, 2021 · d2cc1e9 · d2cc1e9
1 parent e847abd
commit d2cc1e9
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/py/setup.py b/py/setup.py
@@ -171,7 +171,15 @@ def run(self):
 class CleanCommand(Command):
     """Custom clean command to tidy up the project root."""
     PY_CLEAN_FILES = [
-        './build', './dist', './trtorch/__pycache__', './trtorch/lib', './*.pyc', './*.tgz', './*.egg-info'
+        './build',
+        './dist',
+        './trtorch/__pycache__',
+        './trtorch/lib',
+        './trtorch/include',
+        './trtorch/bin',
+        './*.pyc',
+        './*.tgz',
+        './*.egg-info'
     ]
     description = "Command to tidy up the project root"
     user_options = []