Skip to content

Commit 0dce99d

Browse files
authored
hotfix(bnb): install scipy with bitsanbytes to avoid ModuleNotFoundError (#492)
1 parent 355c974 commit 0dce99d

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

.github/workflows/build.yml

+2-2
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@ jobs:
2525
python -m pip install --upgrade pip
2626
pip install -r requirements.txt
2727
# Install extras
28-
# [bnb]
29-
pip install bitsandbytes
28+
# [bnb] (TODO: Remove `scipy` once `bnb` adds it as hard dep)
29+
pip install bitsandbytes scipy
3030
# [dev]
3131
pip install black hypothesis isort flake8 pre-commit pytest pytest-cov
3232

setup.cfg

+3-1
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,9 @@ install_requires =
2929
tritonclient
3030

3131
[options.extras_require]
32-
bnb = bitsandbytes
32+
bnb =
33+
bitsandbytes
34+
scipy # fix(bnb): Remove when `bitsandbytes` adds this dependency
3335
dev =
3436
black
3537
hypothesis

0 commit comments

Comments
 (0)