Clean bazel stuff on distutils clean. (#5274)

* Clean bazel stuff on distutils clean * Fix python formatting
pytorch · Jul 10, 2023 · a2d99ac · a2d99ac
1 parent cba267f
commit a2d99ac
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -163,6 +163,9 @@ def maybe_bundle_libtpu(base_dir):
 
 class Clean(distutils.command.clean.clean):
 
+  def bazel_clean_(self):
+    self.spawn(['bazel', 'clean', '--expunge'])
+
   def run(self):
     import glob
     import re
@@ -183,6 +186,8 @@ def run(self):
             except OSError:
               shutil.rmtree(filename, ignore_errors=True)
 
+    self.execute(self.bazel_clean_, (), msg="Cleaning bazel outputs")
+
     # It's an old-style class in Python 2.7...
     distutils.command.clean.clean.run(self)