diff --git a/src/setuptools_dso/dsocmd.py b/src/setuptools_dso/dsocmd.py index c911892..f59d310 100644 --- a/src/setuptools_dso/dsocmd.py +++ b/src/setuptools_dso/dsocmd.py @@ -11,10 +11,19 @@ import multiprocessing as MP import logging as log -try: - from wheel.bdist_wheel import bdist_wheel as _bdist_wheel -except ImportError: - _bdist_wheel = None +def _import_bdist_wheel(): + global _bdist_wheel + try: + from setuptools.command import bdist_wheel as _bdist_wheel + return + except ImportError: + pass + try: + from wheel.bdist_wheel import bdist_wheel as _bdist_wheel + except ImportError: + _bdist_wheel = None + +_import_bdist_wheel() from setuptools import Command, Distribution, Extension as _Extension from setuptools.command.build_ext import build_ext as _build_ext