use torch nightly so we can test all OS

scikit-hep · Jan 21, 2025 · 138e297 · 138e297
1 parent 07fc591
commit 138e297
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -39,7 +39,7 @@ jobs:
     needs: pre-commit
     strategy:
       matrix:
-        os: [ubuntu-latest] #, macOS-latest, windows-latest]
+        os: [ubuntu-latest, macOS-latest, windows-latest]
         java-version: [17]
         java-distribution: ["corretto"]
         python-version: ["3.9", "3.13"]
@@ -73,7 +73,7 @@ jobs:
         uv pip install --system --upgrade pip setuptools wheel
         # mltool installs
         # c.f. https://github.com/astral-sh/uv/issues/3437
-        python -m pip install torch --index-url https://download.pytorch.org/whl/cpu # torchvision torchaudio
+        python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu # torchvision torchaudio
         uv pip install --system xgboost
         uv pip install --system 'tritonclient[grpc,http]!=2.41.0'
         # install checked out coffea
@@ -86,7 +86,7 @@ jobs:
         uv pip install --system --upgrade pip setuptools wheel
         # mltool installs
         # c.f. https://github.com/astral-sh/uv/issues/3437
-        python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+        python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu # torchvision torchaudio
         uv pip install --system xgboost
         # install checked out coffea
         uv pip install --system -q '.[dev,dask,spark]' --upgrade
@@ -98,10 +98,10 @@ jobs:
         uv pip install --system --upgrade pip setuptools wheel
         # mltool installs
         # c.f. https://github.com/astral-sh/uv/issues/3437
-        python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+        python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu # torchvision torchaudio
         uv pip install --system xgboost
         # install checked out coffea
-        uv pip install --system -q '.[dev,dask]' 'numpy<2' --upgrade
+        uv pip install --system -q '.[dev,dask]' --upgrade
         uv pip list --system
         java -version