Revert to pytorch installation with uv in CI
#5930
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # Runs the complete test suite incl. many external command line dependencies (like Openbabel) | |
| # as well as the pymatgen.ext package. Coverage used to be computed based on this workflow. | |
| name: Tests | |
| on: | |
| push: | |
| branches: [master] | |
| paths-ignore: ["**/*.md", docs/**] | |
| pull_request: | |
| branches: [master] | |
| paths-ignore: ["**/*.md", docs/**] | |
| workflow_dispatch: | |
| workflow_call: # make this workflow reusable by release.yml | |
| permissions: | |
| contents: read | |
| jobs: | |
| test: | |
| # prevent this action from running on forks | |
| if: github.repository == 'materialsproject/pymatgen' | |
| defaults: | |
| run: | |
| shell: bash -l {0} # enables conda/mamba env activation by reading bash profile | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| # pytest-split automatically distributes work load so parallel jobs finish in similar time | |
| os: [ubuntu-latest, windows-latest] | |
| python-version: ["3.9", "3.12"] | |
| split: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] | |
| # include/exclude is meant to maximize CI coverage of different platforms and python | |
| # versions while minimizing the total number of jobs. We run all pytest splits with the | |
| # oldest supported python version (currently 3.9) on windows (seems most likely to surface | |
| # errors) and with newest version (currently 3.12) on ubuntu (to get complete and speedy | |
| # coverage on unix). We ignore mac-os, which is assumed to be similar to ubuntu. | |
| exclude: | |
| - os: windows-latest | |
| python-version: "3.12" | |
| - os: ubuntu-latest | |
| python-version: "3.9" | |
| runs-on: ${{ matrix.os }} | |
| env: | |
| PMG_MAPI_KEY: ${{ secrets.PMG_MAPI_KEY }} | |
| PMG_VASP_PSP_DIR: ${{ github.workspace }}/tests/files | |
| steps: | |
| - name: Check out repo | |
| uses: actions/checkout@v4 | |
| - name: Set up micromamba | |
| uses: mamba-org/setup-micromamba@main | |
| - name: Create mamba environment | |
| run: | | |
| micromamba create -n pmg python=${{ matrix.python-version }} --yes | |
| - name: Install ubuntu-only conda dependencies | |
| if: matrix.os == 'ubuntu-latest' | |
| run: | | |
| micromamba install -n pmg -c conda-forge enumlib packmol bader openbabel openff-toolkit --yes | |
| - name: Install pymatgen and dependencies | |
| run: | | |
| micromamba activate pmg | |
| pip install uv | |
| uv pip install numpy cython | |
| uv pip install --editable '.[dev]' | |
| # TODO remove next line once uv could reliable install torch | |
| pip install --no-deps '.[optional]' | |
| # TODO remove next line installing ase from main branch when FrechetCellFilter is released | |
| uv pip install --upgrade 'git+https://gitlab.com/ase/ase' | |
| - name: pytest split ${{ matrix.split }} | |
| run: | | |
| micromamba activate pmg | |
| pytest --splits 10 --group ${{ matrix.split }} --durations-path tests/files/.pytest-split-durations tests | |
| trigger_atomate2_ci: | |
| needs: test | |
| runs-on: ubuntu-latest | |
| # only run if changes were pushed to master | |
| if: github.ref == 'refs/heads/master' | |
| steps: | |
| - name: Trigger Atomate2 CI | |
| env: | |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} | |
| run: | | |
| curl -L \ | |
| -X POST \ | |
| -H "Accept: application/vnd.github+json" \ | |
| -H "Authorization: Bearer ${{ secrets.GITHUB_TOKEN }}" \ | |
| -H "X-GitHub-Api-Version: 2022-11-28" \ | |
| https://api.github.com/repos/materialsproject/atomate2/dispatches \ | |
| -d '{"event_type": "pymatgen-ci-trigger", "client_payload": {"pymatgen_ref": "${{ github.sha }}"}}' |