diff --git a/py-polars/requirements-ci.txt b/py-polars/requirements-ci.txt index fc6fc8d8b511..9bf4d2a58e7e 100644 --- a/py-polars/requirements-ci.txt +++ b/py-polars/requirements-ci.txt @@ -2,5 +2,7 @@ # 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