Skip to content

Commit

Permalink
inline torch install
Browse files Browse the repository at this point in the history
  • Loading branch information
MarcoGorelli committed Aug 12, 2024
1 parent f27326d commit f294a1e
Show file tree
Hide file tree
Showing 4 changed files with 3 additions and 4 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/benchmark.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/test-coverage.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/test-python.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
1 change: 0 additions & 1 deletion py-polars/requirements-ci.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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

0 comments on commit f294a1e

Please sign in to comment.