Specified environment-name for test.yml

pymc-devs · May 25, 2024 · ec9a93d · ec9a93d
1 parent 9bb0a31
commit ec9a93d
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -123,6 +123,7 @@ jobs:
       - name: Set up Python ${{ matrix.python-version }}
         uses: mamba-org/setup-micromamba@v1
         with:
+          environment-name: pytensor-test
           micromamba-version: "latest"
           init-shell: bash
           post-cleanup: "all"
@@ -139,7 +140,7 @@ jobs:
           echo "id=$MATRIX_ID" >> $GITHUB_OUTPUT
 
       - name: Install dependencies
-        shell: bash -l {0}
+        shell: micromamba-shell {0}
         run: |
           mamba install --yes -q "python~=${PYTHON_VERSION}=*_cpython" mkl numpy scipy pip mkl-service graphviz cython pytest coverage pytest-cov pytest-benchmark pytest-mock sympy
           if [[ $INSTALL_NUMBA == "1" ]]; then mamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}=*_cpython" "numba>=0.57"; fi
@@ -154,7 +155,7 @@ jobs:
           INSTALL_JAX: ${{ matrix.install-jax }}
 
       - name: Run tests
-        shell: bash -l {0}
+        shell: micromamba-shell {0}
         run: |
           if [[ $FAST_COMPILE == "1" ]]; then export PYTENSOR_FLAGS=$PYTENSOR_FLAGS,mode=FAST_COMPILE; fi
           if [[ $FLOAT32 == "1" ]]; then export PYTENSOR_FLAGS=$PYTENSOR_FLAGS,floatX=float32; fi
@@ -191,11 +192,12 @@ jobs:
       - name: Set up Python 3.10
         uses: mamba-org/setup-micromamba@v1
         with:
+          environment-name: pytensor-test
           micromamba-version: "latest"
           init-shell: bash
           post-cleanup: "all"
       - name: Install dependencies
-        shell: bash -l {0}
+        shell: micromamba-shell {0}
         run: |
           mamba install --yes -q -c conda-forge "python~=${PYTHON_VERSION}=*_cpython" mkl numpy scipy pip mkl-service cython pytest "numba>=0.57" jax jaxlib pytest-benchmark
           pip install -e ./
@@ -210,7 +212,7 @@ jobs:
           path: ./cache
           key: ${{ runner.os }}-benchmark
       - name: Run benchmarks
-        shell: bash -l {0}
+        shell: micromamba-shell {0}
         run: |
           export PYTENSOR_FLAGS=mode=FAST_COMPILE,warn__ignore_bug_before=all,on_opt_error=raise,on_shape_error=raise,gcc__cxxflags=-pipe
           python -m pytest --runslow --benchmark-only --benchmark-json output.json