Build & Clean: In Setuptools

setup.py

@@ -5,8 +5,6 @@
 # Authors: Axel Huebl
 # License: BSD-3-Clause-LBNL
 #
-from distutils.command.build import build
-from distutils.command.clean import clean
 import os
 import platform
 import re
@@ -16,6 +14,7 @@
 
 from setuptools import Extension, setup
 from setuptools.command.build_ext import build_ext
+from setuptools.command.build_py import build_py as build
 
 
 class CopyPreBuild(build):
@@ -32,10 +31,8 @@ def run(self):
         #   by default, this stays around. We want to make sure generated
         #   files like amrex_*d_pybind.*.(so|pyd) are always only the
         #   ones we want to package and not ones from an earlier wheel's stage
-        c = clean(self.distribution)
-        c.all = True
-        c.finalize_options()
-        c.run()
+        if os.path.exists(self.build_base):
+            shutil.rmtree(self.build_base)
 
         # call superclass
         build.run(self)