JAX 0.4.33 changes how errors work, which broke the nice error_if messages. Hopefully this should be cross-version compatible. #1001

	name: Run tests

	on:
	pull_request:

	jobs:
	run-test:
	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 -r ./tests/requirements.txt

	- name: Checks with pre-commit
	uses: pre-commit/[email protected]

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

Provide feedback