ci-build.yaml

name: CI

on:
  # Trigger the workflow on push or pull request,
  # only on the main branch
  push:
    branches:
      - main
  pull_request:
    branches:
      - main
  workflow_dispatch:

jobs:
  lint_and_typecheck:
    runs-on: ubuntu-latest
    strategy:
      matrix:
        python-version: ['3.11']

    steps:
    - uses: actions/checkout@v4
    - name: Set up Python ${{ matrix.python-version }}
      uses: actions/setup-python@v5
      with:
        python-version: ${{ matrix.python-version }}
    - name: Install dependencies
      run: |
        pip install -e '.[testing]'
    - name: Lint with flake8
      run: |
        flake8 .
    - name: Lint with pylint
      run: |
        pylint --fail-under 9.75 ferminet
    - name: Type check with pytype
      run: |
        pytype ferminet
  build:
    name: "build ${{ matrix.name-prefix }} (py ${{ matrix.python-version }} on ${{ matrix.os }})"
    runs-on: ${{ matrix.os }}
    strategy:
      matrix:
        include:
          - name-prefix: "all tests"
            python-version: '3.10'
            os: ubuntu-latest
            package-overrides: "none"
          - name-prefix: "all tests"
            python-version: '3.11'
            os: ubuntu-latest
            package-overrides: "none"
    steps:
    - uses: actions/checkout@v4
    - name: Set up Python ${{ matrix.python-version }}
      uses: actions/setup-python@v5
    - name: Install dependencies
      run: |
        pip install -e '.[testing]'
        if [ ${{ matrix.package-overrides }} != none ]; then
          pip install ${{ matrix.package-overrides }}
        fi
    - name: Run tests
      run: |
        python -m pytest
    - name: Run multi-device tests
      if: matrix.python-version == '3.11'
      run: |
        FERMINET_CHEX_N_CPU_DEVICES=2 python -m pytest ferminet/tests/train_test.py