enable macos unit tests

PiperOrigin-RevId: 706845110

.github/workflows/nightly_release.yml

@@ -20,6 +20,7 @@ jobs:
     uses: ./.github/workflows/unittests_python.yml
     with:
       trigger-sha: ${{ github.sha }}
+      run-on-macos: true
 
   build-and-release-nightly:
     needs: run-unittests-python

.github/workflows/nightly_unittests.yml

@@ -20,3 +20,4 @@ jobs:
     uses: ./.github/workflows/unittests_python.yml
     with:
       trigger-sha: ${{ github.sha }}
+      run-on-macos: true

.github/workflows/run_post_merge.yml

@@ -28,3 +28,4 @@ jobs:
     uses: ./.github/workflows/unittests_python.yml
     with:
       trigger-sha: ${{ github.event.after }}
+      run-on-macos: false

.github/workflows/run_pre_merge.yml

@@ -64,6 +64,7 @@ jobs:
     uses: ./.github/workflows/unittests_python.yml
     with:
       trigger-sha: ${{ github.event.pull_request.head.sha }}
+      run-on-macos: false
 
   run-model-coverage:
     name: Model Coverage

.github/workflows/unittests_python.yml

@@ -12,16 +12,34 @@ on:
   workflow_call:
     inputs:
       trigger-sha:
-        required: true
         type: string
-
+        required: true
+      run-on-macos:
+        type: boolean
+        default: false
 jobs:
+  prepare-matrix:
+    runs-on: ubuntu-latest
+    outputs:
+      os: ${{ steps.set-matrix.outputs.os }}
+    steps:
+      - name: Set matrix
+        id: set-matrix
+        run: |
+          os=("Linux_runner_8_core")
+          if [[ "${{ inputs.run-on-macos }}" == "true" ]]; then
+            os+=("macos-latest")
+          fi
+          os=$(jq -c -n '$ARGS.positional' --args "${os[@]}")
+          echo "os=${os}"
+          echo "os=${os}" >> $GITHUB_OUTPUT
   test:
+    needs: prepare-matrix
     strategy:
       matrix:
         python-version: ["3.10", "3.11"]
-    runs-on:
-      labels: Linux_runner_8_core
+        os: ${{ fromJson(needs.prepare-matrix.outputs.os) }}
+    runs-on: ${{ matrix.os }}
     steps:
       - uses: actions/checkout@v3
         with:

requirements.txt

@@ -1,9 +1,6 @@
--f https://download.pytorch.org/whl/torch/
-torch==2.5.1+cpu
--f https://download.pytorch.org/whl/torchvision/
-torchvision==0.20.1+cpu
--f https://download.pytorch.org/whl/torchaudio/
-torchaudio==2.5.1+cpu
+torch==2.5.1
+torchvision==0.20.1
+torchaudio==2.5.1
 --pre
 tf-nightly>=2.19.0.dev20241201
 ai-edge-litert-nightly