[dev] remove build artifacts for cleanup (dmlc#7166)

ska278 · Feb 28, 2024 · 5f653c3 · 5f653c3
1 parent ca35563
commit 5f653c3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/python/setup.py b/python/setup.py
@@ -84,6 +84,13 @@ def cleanup():
             remove_lib("dgl_sparse")
             remove_lib("graphbolt")
 
+    # Remove build artifacts.
+    dir_to_remove = ["build", "dist", "dgl.egg-info"]
+    for dir_ in dir_to_remove:
+        print(f"Removing {dir_}")
+        if os.path.isdir(dir_):
+            shutil.rmtree(dir_)
+
 
 def config_cython():
     """Try to configure cython and return cython configuration"""