From 2cd9a1a6608bd3c87b31630d977ad0bb5f8514f1 Mon Sep 17 00:00:00 2001
From: Per-Arne Andersen <per@sysx.no>
Date: Mon, 28 Aug 2023 16:25:15 +0200
Subject: [PATCH] Update setup.py

---
 setup.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index 2f88b519..1ae49d29 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 from setuptools import setup, find_packages
 from setuptools.command.develop import develop
 from setuptools.command.install import install
-from setuptools.command.bdist_wheel import bdist_wheel
+from setuptools.command.bdist import bdist
 import cffi
 import tomli
 from typing import Dict, Any
@@ -45,7 +45,7 @@ def build_cffi():
 
     ffibuilder.compile(verbose=True)
 
-class TMUbdistWheel(bdist_wheel):
+class TMUbdistWheel(bdist):
     """
     Custom install command that builds the CFFI modules
     before proceeding with the standard installation process.
@@ -89,7 +89,7 @@ def has_ext_modules(foo):
     cmdclass={
         "install": TMUInstall,
         "develop": TMUDevelop,
-        "bdist_wheel": TMUbdistWheel
+        "bdist": TMUbdistWheel
     },
     distclass=BinaryDistribution
 )