From 562b0d36473d981cd65a74e1413769cc108f4cd8 Mon Sep 17 00:00:00 2001 From: Ricardo Vieira Date: Tue, 11 Nov 2025 11:45:47 +0100 Subject: [PATCH 1/2] Move Pytorch __init__ to right folder --- pytensor/link/__init__.py | 1 - pytensor/link/pytorch/__init__.py | 1 + 2 files changed, 1 insertion(+), 1 deletion(-) create mode 100644 pytensor/link/pytorch/__init__.py diff --git a/pytensor/link/__init__.py b/pytensor/link/__init__.py index c8c236a854..e69de29bb2 100644 --- a/pytensor/link/__init__.py +++ b/pytensor/link/__init__.py @@ -1 +0,0 @@ -from pytensor.link.pytorch.linker import PytorchLinker diff --git a/pytensor/link/pytorch/__init__.py b/pytensor/link/pytorch/__init__.py new file mode 100644 index 0000000000..c8c236a854 --- /dev/null +++ b/pytensor/link/pytorch/__init__.py @@ -0,0 +1 @@ +from pytensor.link.pytorch.linker import PytorchLinker From abe66bf3b02b897d879d3418a5c5b00f42aae7f9 Mon Sep 17 00:00:00 2001 From: Ricardo Vieira Date: Sun, 16 Nov 2025 14:02:14 +0100 Subject: [PATCH 2/2] Remove mkl pin from torch CI --- .github/workflows/test.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 8a28d79393..b2f6584687 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -204,7 +204,7 @@ jobs: fi if [[ $INSTALL_NUMBA == "1" ]]; then micromamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}" "numba>=0.57"; fi if [[ $INSTALL_JAX == "1" ]]; then micromamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}" jax jaxlib numpyro equinox && pip install tfp-nightly; fi - if [[ $INSTALL_TORCH == "1" ]]; then micromamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}" pytorch pytorch-cuda=12.1 "mkl<=2024.0" -c pytorch -c nvidia; fi + if [[ $INSTALL_TORCH == "1" ]]; then micromamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}" pytorch pytorch-cuda=12.1 -c pytorch -c nvidia; fi if [[ $INSTALL_MLX == "1" ]]; then micromamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}" mlx; fi if [[ $INSTALL_XARRAY == "1" ]]; then micromamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}" xarray xarray-einstats; fi