examples.yaml

name: examples

on:
  push:
    branches: [ main ]
  pull_request:
    branches: [ main ]

jobs:
  precommit:
    name: Pre-commit checks
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v3

  examples:
    runs-on: ubuntu-latest
    needs:
      - precommit
    strategy:
      matrix:
        python-version: [3.11]
    steps:
      - uses: actions/checkout@v3
      - name: Set up Python ${{ matrix.python-version }}
        uses: actions/setup-python@v3
        with:
          python-version: ${{ matrix.python-version }}
      - name: Install dependencies
        run: |
          pip install hatch matplotlib
      - name: Build package
        run: |
          pip install jaxlib jax
          pip install .
      - name: Run tests
        run: |
            python examples/autoregressive_inference_surjection.py --n-iter 10
            python examples/conditional_density_estimation.py --n-iter 10 --model coupling
            python examples/conditional_density_estimation.py --n-iter 10 --model autoregressive
            python examples/coupling_inference_surjection.py --n-iter 10