remove the egg-info dir

ROCm · Oct 1, 2024 · c62d4b4 · c62d4b4
1 parent 5a50530
commit c62d4b4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/build/rocm/tools/build_wheels.py b/build/rocm/tools/build_wheels.py
@@ -256,6 +256,7 @@ def main():
             shutil.copy(whl, wheelhouse_dir)
     # delete the 'dist' directory since it causes permissions issues
     shutil.rmtree(os.path.join(args.jax_path, "dist"))
+    shutil.rmtree(os.path.join(args.jax_path, "jax.egg-info"))
 
 
 if __name__ == "__main__":