diff --git a/.github/workflows/run_tests.yml b/.github/workflows/run_tests.yml index 7afd53a8..61288e8d 100644 --- a/.github/workflows/run_tests.yml +++ b/.github/workflows/run_tests.yml @@ -43,12 +43,17 @@ jobs: flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics # # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide # flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics - - name: Install dependencies - run: | - $CONDA/bin/conda env update --file environment_all_backends.yml --name base + - name: provision-with-micromamba + uses: mamba-org/provision-with-micromamba@main + with: + environment-file: environment_all_backends.yml + environment-name: torchquad + cache-downloads: true - name: Test with pytest + shell: bash -l {0} run: | + micromamba activate torchquad cd torchquad/tests - conda install pytest - $CONDA/bin/pip3 install pytest-error-for-skips - $CONDA/bin/pytest -ra --error-for-skips + pip install pytest + pip install pytest-error-for-skips + pytest -ra --error-for-skips diff --git a/environment_all_backends.yml b/environment_all_backends.yml index a902b61e..630658bb 100644 --- a/environment_all_backends.yml +++ b/environment_all_backends.yml @@ -15,9 +15,9 @@ dependencies: - tqdm>=4.56.0 # Numerical backend installations with CUDA support where possible: - numpy>=1.19.5 -- cudatoolkit>=11.1 +- cudatoolkit>=11.8 - pytorch>=1.9=*cuda* -- tensorflow-gpu>=2.6.0 +- tensorflow-gpu>=2.10.0 # jaxlib with CUDA support is not available for conda - pip: - --find-links https://storage.googleapis.com/jax-releases/jax_releases.html