run_tests.yml

name: Run tests

on:
  pull_request:

jobs:
  run-tests:
    strategy:
      matrix:
        python-version: [ 3.9, 3.11 ]
        os: [ ubuntu-latest ]
      fail-fast: false
    runs-on: ${{ matrix.os }}
    steps:
      - name: Checkout code
        uses: actions/checkout@v2

      - name: Set up Python ${{ matrix.python-version }}
        uses: actions/setup-python@v2
        with:
          python-version: ${{ matrix.python-version }}

      - name: Install dependencies
        run: |
          python -m pip install --upgrade pip
          python -m pip install pytest wheel scipy numpy optax jaxlib

      - name: Checks with pre-commit
        uses: pre-commit/action@v2.0.3

      - name: Test with pytest
        run: |
          python -m pip install .
          python -m test