Deprecate distrax and update examples #142

	name: ci

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

	jobs:
	precommit:
	name: Pre-commit checks
	runs-on: ubuntu-latest
	steps:
	- uses: actions/checkout@v3
	- uses: actions/setup-python@v3
	- uses: pre-commit/[email protected]

	build:
	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
	- name: Build package
	run: \|
	hatch build

	lints:
	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
	- name: Run lints
	run: \|
	hatch run test:lint

	tests:
	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
	- name: Build package
	run: \|
	pip install jaxlib==0.4.24 jax==0.4.24
	- name: Run tests
	run: \|
	hatch run test:test
	- name: Upload coverage reports to Codecov
	uses: codecov/codecov-action@v3
	env:
	CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}

Provide feedback