Merge pull request #380 from scverse/Ori-Adding-MPS-MACOS-local-tests

test: Added localy manual macos tests for mps

.github/workflows/run_notebook_individual_macos.yaml

@@ -0,0 +1,58 @@
+name: run notebook MACOS MPS (individual)
+
+on:
+  workflow_dispatch:
+    inputs:
+      notebook:
+        description: "Path to the notebook to run"
+        required: true
+        type: choice
+        options: # not including R tutorials currently
+          - atac/PeakVI.ipynb
+          - atac/PoissonVI.ipynb
+          - atac/scbasset_batch.ipynb
+          - atac/scbasset.ipynb
+          - dev/data_tutorial.ipynb
+          - dev/model_user_guide.ipynb
+          - dev/module_user_guide.ipynb
+          - hub/cellxgene_census_model.ipynb
+          - hub/minification.ipynb
+          - hub/scvi_hub_intro_and_download.ipynb
+          - hub/scvi_hub_upload_and_large_files.ipynb
+          - multimodal/cite_scrna_integration_w_totalVI.ipynb
+          - multimodal/MultiVI_tutorial.ipynb
+          - multimodal/totalVI_reference_mapping.ipynb
+          - multimodal/totalVI.ipynb
+          - quick_start/api_overview.ipynb
+          - quick_start/data_loading.ipynb
+          - scbs/MethylVI_batch.ipynb
+          - scrna/amortized_lda.ipynb
+          - scrna/AutoZI_tutorial.ipynb
+          - scrna/cellassign_tutorial.ipynb
+          - scrna/contrastiveVI_tutorial.ipynb
+          - scrna/harmonization.ipynb
+          - scrna/linear_decoder.ipynb
+          - scrna/query_hlca_knn.ipynb
+          - scrna/scanvi_fix.ipynb
+          - scrna/scarches_scvi_tools.ipynb
+          - scrna/scVI_DE_worm.ipynb
+          - scrna/seed_labeling.ipynb
+          - scrna/tabula_muris.ipynb
+          - scrna/MrVI_tutorial.ipynb
+          - spatial/cell2location_lymph_node_spatial_tutorial.ipynb
+          - spatial/DestVI_tutorial.ipynb
+          - spatial/gimvi_tutorial.ipynb
+          - spatial/stereoscope_heart_LV_tutorial.ipynb
+          - spatial/tangram_scvi_tools.ipynb
+          - tuning/autotune_scvi.ipynb
+      version:
+        description: "latest/stable/semver/branch"
+        required: true
+        type: string
+
+jobs:
+  notebook:
+    uses: ./.github/workflows/run_notebook_macos.yaml
+    with:
+      notebook: ${{ inputs.notebook }}
+      version: ${{ inputs.version }}

.github/workflows/run_notebook_macos.yaml

@@ -0,0 +1,51 @@
+name: run notebook
+
+on:
+  workflow_call:
+    inputs:
+      notebook:
+        description: "Path to the notebook to run"
+        required: true
+        type: string
+      version:
+        description: "latest/stable/semver/branch"
+        required: true
+        type: string
+
+jobs:
+  notebook:
+    runs-on: [self-hosted, macOS, X64, MPS]
+
+    name: macos_integration
+
+    timeout-minutes: 600 # lenient timeout for scbasset tutorial
+
+    env:
+      OS: ${{ matrix.os }}
+      PYTHON: ${{ matrix.python }}
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Create Conda environment and install dependencies
+        run: |
+          conda init bash
+          source ~/.bash_profile
+          conda activate scvi
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip wheel uv
+          python -m pip install "scvi-tools[tests]"
+          python -m pip install jax-metal
+          python -m ipykernel install
+
+      - name: Run notebook
+        run: |
+          python -m jupyter nbconvert --execute --inplace ${{ inputs.notebook }}
+
+      - uses: peter-evans/create-pull-request@v5
+        with:
+          commit-message: run MPS ${{ inputs.notebook }}
+          title: automated MPS update for ${{ inputs.notebook }}
+          branch: update-mps-${{ inputs.notebook }}