Add support for PyMC v5 (and fix some recent Jax issues) #378

	name: Python
	on:
	push:
	branches:
	- main
	tags:
	- "*"
	paths-ignore:
	- "docs/**"
	pull_request:

	concurrency:
	group: ${{ github.workflow }}-${{ github.head_ref }}
	cancel-in-progress: true

	jobs:
	tests:
	runs-on: ${{ matrix.os }}
	strategy:
	fail-fast: false
	matrix:
	python-version: ["3.10"]
	os: ["ubuntu-latest"]
	session:
	- "core"
	- "jax"
	- "pymc3"
	- "pymc"
	- "pymc_jax"

	steps:
	- name: Checkout
	uses: actions/checkout@v2
	with:
	fetch-depth: 0
	submodules: true

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

	- name: Install dependencies
	run: \|
	python -m pip install -U pip
	python -m pip install -U coveralls coverage[toml] nox

	- name: Run tests
	run: \|
	python -m nox --non-interactive --error-on-missing-interpreter \
	--session ${{ matrix.session }}-${{ matrix.python-version }}

	- name: Combine and upload coverage
	run: \|
	python -m coverage combine
	python -m coverage xml -i
	python -m coveralls --service=github
	env:
	GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
	COVERALLS_PARALLEL: true
	COVERALLS_FLAG_NAME: ${{matrix.os}}-${{matrix.session}}-${{matrix.python-version}}

	coverage:
	needs: tests
	runs-on: ubuntu-latest
	steps:
	- name: Setup Python
	uses: actions/setup-python@v2
	with:
	python-version: "3.9"
	- name: Finish coverage collection
	run: \|
	python -m pip install -U pip
	python -m pip install -U coveralls
	python -m coveralls --finish
	env:
	GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

Provide feedback