Test with Metal backend

.github/workflows/test_linux_pre.yml

@@ -14,7 +14,7 @@ concurrency:
 
 jobs:
   test:
-    # if PR has label "cuda tests" or "all tests" or if scheduled or manually triggered
+    # if PR has label "prerelease tests" or "all tests" or if scheduled or manually triggered
     if: >-
       (
         contains(github.event.pull_request.labels.*.name, 'prerelease tests') ||

.github/workflows/test_macos_m1.yml

@@ -1,6 +1,9 @@
 name: Test (MacOS M1)
 
 on:
+  pull_request:
+    branches: [main, 1.1.x]
+    types: [labeled, synchronize, opened]
   schedule:
     - cron: "0 10 * * *" # runs at 10:00 UTC -> 03:00 PST every day
   workflow_dispatch:
@@ -11,6 +14,15 @@ concurrency:
 
 jobs:
   test:
+    # if PR has label "metal tests" or "all tests" or "platform tests" if scheduled or manually triggered
+    if: >-
+      (
+        contains(github.event.pull_request.labels.*.name, 'metal tests') ||
+        contains(github.event.pull_request.labels.*.name, 'platform tests') ||
+        contains(github.event.pull_request.labels.*.name, 'all tests') ||
+        contains(github.event_name, 'schedule') ||
+        contains(github.event_name, 'workflow_dispatch')
+      )
     runs-on: ${{ matrix.os }}
     defaults:
       run:
@@ -41,6 +53,10 @@ jobs:
         run: |
           python -m pip install --upgrade pip wheel
 
+      - name: Install Metal dependencies
+        run: |
+          pip install -U jax-metal numpy wheel ml-dtypes torch torchvision torchaudio
+
       - name: Install dependencies
         run: |
           pip install ".[tests]"

scvi/model/_utils.py

@@ -108,11 +108,6 @@ def parse_device_args(
             UserWarning,
             stacklevel=settings.warnings_stacklevel,
         )
-    elif _accelerator == "mps" and accelerator == "auto":
-        # auto accelerator should not default to mps
-        connector = _AcceleratorConnector(accelerator="cpu", devices=devices)
-        _accelerator = connector._accelerator_flag
-        _devices = connector._devices_flag
     elif _accelerator == "mps" and accelerator != "auto":
         warnings.warn(
             "`accelerator` has been set to `mps`. Please note that not all PyTorch "