delete the dist directory

ROCm · Oct 1, 2024 · 5a50530 · 5a50530
1 parent 0bf3f5d
commit 5a50530
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/build/rocm/tools/build_wheels.py b/build/rocm/tools/build_wheels.py
@@ -254,6 +254,8 @@ def main():
         if os.path.basename(whl).startswith("jax-"):
             LOG.info("Copying %s into %s" % (whl, wheelhouse_dir))
             shutil.copy(whl, wheelhouse_dir)
+    # delete the 'dist' directory since it causes permissions issues
+    shutil.rmtree(os.path.join(args.jax_path, "dist"))
 
 
 if __name__ == "__main__":