Skip to content

Commit 5c1639d

Browse files
authored
add rsc (#21)
1 parent 0124a5b commit 5c1639d

3 files changed

Lines changed: 21 additions & 2 deletions

File tree

.cirun.yml

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
runners:
2+
- name: aws-gpu-runner
3+
cloud: aws
4+
instance_type: g4dn.xlarge
5+
machine_image: ami-067a4ba2816407ee9
6+
region: eu-north-1
7+
preemptible:
8+
- true
9+
- false
10+
labels:
11+
- cirun-aws-gpu

.github/workflows/integration-test.yml

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,15 +32,16 @@ jobs:
3232
- {name: "pertpy", extras: "de"}
3333
- {name: "decoupler", extras: ""}
3434
- {name: "SnapATAC2", extras: ""}
35+
- {name: "rapids_singlecell", extras: "rapids12", gpu: true}
3536
exclude:
3637
- { python: 3.12, is_pre: true }
3738

3839
defaults:
3940
run:
4041
shell: bash -el {0}
4142

42-
runs-on: ubuntu-latest
43-
43+
runs-on: ${{ matrix.package.gpu && format('cirun-aws-gpu--{0}', github.run_id) || 'ubuntu-latest' }}
44+
4445
env:
4546
# This env variable is used by the `gh` CLI
4647
GH_TOKEN: ${{ secrets.TOKEN_FOR_ISSUE_WRITE }}
@@ -60,6 +61,10 @@ jobs:
6061
filter: blob:none
6162
path: ${{ matrix.package.name }}
6263

64+
- name: Nvidia SMI sanity check
65+
if: matrix.package.gpu
66+
run: nvidia-smi
67+
6368
- name: Install Python
6469
uses: actions/setup-python@v5
6570
with:
@@ -77,6 +82,8 @@ jobs:
7782
uv pip install --system toml-cli
7883
7984
- name: Install AnnData and scverse package
85+
env:
86+
UV_EXTRA_INDEX_URL: ${{ matrix.package.gpu && 'https://pypi.nvidia.com' || '' }}
8087
run: |
8188
EXTRAS=""
8289
GROUP=""

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ The following packages are tested:
2020
- pertpy
2121
- decoupler
2222
- SnapATAC2
23+
- rapids-singlecell
2324

2425
## How it Works
2526

0 commit comments

Comments
 (0)