Skip to content

Commit aeffdf8

Browse files
committed
fix: add missing ml dependencies to ci workflow
1 parent 83fb324 commit aeffdf8

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

.github/workflows/ci.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,9 @@ jobs:
2222
- name: Install dependencies
2323
run: |
2424
python -m pip install --upgrade pip
25-
pip install pytest httpx fastapi pydantic-settings pandas scikit-learn datasets transformers
25+
pip install pytest httpx fastapi pydantic-settings pandas scikit-learn datasets transformers peft bitsandbytes torch --extra-index-url https://download.pytorch.org/whl/cpu
2626
2727
- name: Run tests
2828
run: |
29+
export BITSANDBYTES_NOWELCOME=1
2930
python -m pytest tests/test_data.py tests/test_api.py

0 commit comments

Comments
 (0)