Skip to content

Commit

Permalink
Fixed formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
rickardp committed Feb 4, 2024
1 parent eba19a8 commit 3288a0f
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@
from setuptools import find_packages, setup
from setuptools.dist import Distribution


libs = list(glob.glob("./bitsandbytes/libbitsandbytes*.*"))
libs = [os.path.basename(p) for p in libs]
print("libs:", libs)
Expand All @@ -17,11 +16,13 @@
def read(fname):
return open(os.path.join(os.path.dirname(__file__), fname)).read()


# Tested with wheel v0.29.0
class BinaryDistribution(Distribution):
def has_ext_modules(self):
return True


setup(
name="bitsandbytes",
version="0.43.0.dev0",
Expand All @@ -33,16 +34,16 @@ def has_ext_modules(self):
url="https://github.com/TimDettmers/bitsandbytes",
packages=find_packages(),
package_data={"": libs},
install_requires=['torch', 'numpy'],
install_requires=["torch", "numpy"],
extras_require={
'benchmark': ['pandas', 'matplotlib'],
'test': ['scipy'],
"benchmark": ["pandas", "matplotlib"],
"test": ["scipy"],
},
long_description=read("README.md"),
long_description_content_type="text/markdown",
classifiers=[
"Development Status :: 4 - Beta",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
],
distclass=BinaryDistribution
distclass=BinaryDistribution,
)

0 comments on commit 3288a0f

Please sign in to comment.