Add RAPIDS-Singlecell

.cirun.yml

@@ -0,0 +1,11 @@
+runners:
+  - name: aws-gpu-runner
+    cloud: aws
+    instance_type: g4dn.xlarge
+    machine_image: ami-067a4ba2816407ee9
+    region: eu-north-1
+    preemptible:
+      - true
+      - false
+    labels:
+      - cirun-aws-gpu

.github/workflows/integration-test.yml

@@ -32,15 +32,16 @@ jobs:
           - {name: "pertpy", extras: "de"}
           - {name: "decoupler", extras: ""}
           - {name: "SnapATAC2", extras: ""}
+          - {name: "rapids_singlecell", extras: "rapids12", gpu: true}
         exclude:
           - { python: 3.12, is_pre: true }
 
     defaults:
       run:
         shell: bash -el {0}
 
-    runs-on: ubuntu-latest
-  
+    runs-on: ${{ matrix.package.gpu && format('cirun-aws-gpu--{0}', github.run_id) || 'ubuntu-latest' }}
+
     env:
       # This env variable is used by the `gh` CLI
       GH_TOKEN: ${{ secrets.TOKEN_FOR_ISSUE_WRITE }}
@@ -60,6 +61,10 @@ jobs:
           filter: blob:none
           path: ${{ matrix.package.name }}
 
+      - name: Nvidia SMI sanity check
+        if: matrix.package.gpu
+        run: nvidia-smi
+
       - name: Install Python
         uses: actions/setup-python@v5
         with:
@@ -77,6 +82,8 @@ jobs:
           uv pip install --system toml-cli
 
       - name: Install AnnData and scverse package
+        env:
+          UV_EXTRA_INDEX_URL: ${{ matrix.package.gpu && 'https://pypi.nvidia.com' || '' }}
         run: |
           EXTRAS=""
           GROUP=""

README.md

@@ -20,6 +20,7 @@ The following packages are tested:
 - pertpy
 - decoupler
 - SnapATAC2
+- rapids-singlecell
 
 ## How it Works