TEST: Check torch with MPS devices fix

explosion · Nov 19, 2023 · 7ba697c · 7ba697c
1 parent c16f552
commit 7ba697c
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -45,21 +45,21 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        os: [ubuntu-latest, windows-latest, macos-latest]
+        os: [macos-latest]
         python_version: ["3.12"]
         include:
-          - os: windows-2019
-            python_version: "3.6"
-          - os: macos-latest
-            python_version: "3.7"
-          - os: ubuntu-latest
-            python_version: "3.8"
-          - os: windows-latest
-            python_version: "3.9"
+#          - os: windows-2019
+#            python_version: "3.6"
+#          - os: macos-latest
+#            python_version: "3.7"
+#          - os: ubuntu-latest
+#            python_version: "3.8"
+#          - os: windows-latest
+#            python_version: "3.9"
           - os: macos-latest
             python_version: "3.10"
-          - os: ubuntu-latest
-            python_version: "3.11"
+#          - os: ubuntu-latest
+#            python_version: "3.11"
 
     runs-on: ${{ matrix.os }}
     env:
@@ -141,7 +141,7 @@ jobs:
           # there is a bug related to MPS devices in github macos runners that
           # will be fixed in torch v2.1.1
           # https://github.com/pytorch/pytorch/pull/111576
-          pip install "torch>=2.1.1; sys_platform=='darwin'" --extra-index-url https://download.pytorch.org/whl/cpu
+          pip install "torch>=2.1.1; sys_platform=='darwin'"
           pip install "numpy~=1.23.0; python_version=='3.10' and sys_platform=='win32'"
           pip install "numpy<1.24.0"
           pip install -r requirements.txt