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)