diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml index 335d0a32b754..e7095e35820f 100644 --- a/.github/workflows/benchmark.yml +++ b/.github/workflows/benchmark.yml @@ -48,7 +48,7 @@ jobs: - name: Install Python dependencies working-directory: py-polars - run: uv pip install --compile-bytecode -r requirements-dev.txt -r requirements-ci.txt --verbose + run: uv pip install --compile-bytecode -r requirements-dev.txt --extra-index-url https://download.pytorch.org/whl/cpu torch -r requirements-ci.txt --verbose - name: Set up Rust run: rustup show diff --git a/.github/workflows/test-coverage.yml b/.github/workflows/test-coverage.yml index ce642bbec306..c2c2d26bfc41 100644 --- a/.github/workflows/test-coverage.yml +++ b/.github/workflows/test-coverage.yml @@ -103,7 +103,7 @@ jobs: - name: Install Python dependencies working-directory: py-polars - run: uv pip install --compile-bytecode -r requirements-dev.txt -r requirements-ci.txt --verbose + run: uv pip install --compile-bytecode -r requirements-dev.txt --extra-index-url https://download.pytorch.org/whl/cpu torch -r requirements-ci.txt --verbose - name: Set up Rust run: rustup component add llvm-tools-preview diff --git a/.github/workflows/test-python.yml b/.github/workflows/test-python.yml index ca717ef191f2..fd4a31774342 100644 --- a/.github/workflows/test-python.yml +++ b/.github/workflows/test-python.yml @@ -66,7 +66,7 @@ jobs: - name: Install Python dependencies run: | pip install uv - uv pip install --compile-bytecode -r requirements-dev.txt -r requirements-ci.txt --verbose + uv pip install --compile-bytecode -r requirements-dev.txt --extra-index-url https://download.pytorch.org/whl/cpu torch -r requirements-ci.txt --verbose - name: Set up Rust run: rustup show diff --git a/py-polars/requirements-ci.txt b/py-polars/requirements-ci.txt index 050e4e259084..fc6fc8d8b511 100644 --- a/py-polars/requirements-ci.txt +++ b/py-polars/requirements-ci.txt @@ -2,6 +2,5 @@ # Packages that we require for unit tests that run on CI # (installable via `make requirements-all`) # ------------------------------------------------------- ---extra-index-url https://download.pytorch.org/whl/cpu torch jax[cpu] pyiceberg>=0.5.0