diff --git a/setup.py b/setup.py index 7a82b7717..381cdb18e 100644 --- a/setup.py +++ b/setup.py @@ -5,7 +5,7 @@ import glob import os -from setuptools import find_packages, setup +from setuptools import Extension, find_packages, setup libs = list(glob.glob("./bitsandbytes/libbitsandbytes*.so")) @@ -36,6 +36,9 @@ def read(fname): }, long_description=read("README.md"), long_description_content_type="text/markdown", + # HACK: pretend we have a native extension module so the wheel is tagged + # correctly with a platform tag (e.g. `-linux_x86_64.whl`). + ext_modules=[Extension("bitsandbytes", sources=[], language="c")], classifiers=[ "Development Status :: 4 - Beta", "Topic :: Scientific/Engineering :: Artificial Intelligence",