Skip to content

Commit 8ba2abb

Browse files
committed
adjust workflow to install mamba-ssm when running tests on torch backend
1 parent 62cf5b5 commit 8ba2abb

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

.github/workflows/tests.yaml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,13 @@ jobs:
6161
run: |
6262
pip install -U torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
6363
64+
# we currently have extra dependencies for torch tests
65+
# TODO: can we install this dynamically? should we add it to the test dependencies?
66+
- name: Install Extra Dependencies
67+
if: ${{ matrix.backend == 'torch' }}
68+
run: |
69+
pip install -U mamba-ssm
70+
6471
- name: Show Environment Info
6572
run: |
6673
python --version

0 commit comments

Comments
 (0)