address feedback

Author: jessegrabowski

.github/workflows/test.yml

@@ -84,7 +84,7 @@ jobs:
         install-mlx: [0]
         install-xarray: [0]
         part:
-          - "tests --ignore=tests/tensor --ignore=tests/scan --ignore=tests/xtensor --ignore=tests/link/mlx"
+          - "tests --ignore=tests/tensor --ignore=tests/scan --ignore=tests/xtensor"
           - "tests/scan"
           - "tests/tensor --ignore=tests/tensor/rewriting --ignore=tests/tensor/test_math.py --ignore=tests/tensor/test_basic.py --ignore=tests/tensor/test_inplace.py --ignore=tests/tensor/conv --ignore=tests/tensor/test_blas.py --ignore=tests/tensor/test_elemwise.py --ignore=tests/tensor/test_math_scipy.py"
           - "tests/tensor/rewriting"
@@ -153,7 +153,6 @@ jobs:
             part: "tests/xtensor"
           - os: "macos-15"
             python-version: "3.11"
-            numpy-version: ">=2.0"
             fast-compile: 0
             float32: 0
             install-mlx: 1
@@ -181,7 +180,7 @@ jobs:
           environment-name: pytensor-test
           micromamba-version: "1.5.10-0" # until https://github.com/mamba-org/setup-micromamba/issues/225 is resolved
           init-shell: bash
-          post-cleanup: "none"
+          post-cleanup: "all"
           create-args: python=${{ matrix.python-version }}
 
       - name: Create matrix id
@@ -208,8 +207,7 @@ jobs:
           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_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
-          pip install pytest-sphinx
-          
+
           pip install -e ./
           micromamba list && pip freeze
           python -c 'import pytensor; print(pytensor.config.__str__(print_doc=False))'
@@ -269,7 +267,7 @@ jobs:
           environment-name: pytensor-test
           micromamba-version: "1.5.10-0" # until https://github.com/mamba-org/setup-micromamba/issues/225 is resolved
           init-shell: bash
-          post-cleanup: "none"
+          post-cleanup: "all"
       - name: Install dependencies
         shell: micromamba-shell {0}
         run: |