Update dependencies and use new jax random key #1359

	name: Tests

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

	jobs:
	style:
	name: Check the code style
	runs-on: ubuntu-latest
	steps:
	- uses: actions/checkout@v2
	- uses: actions/setup-python@v2
	with:
	python-version: 3.9
	- uses: pre-commit/[email protected]

	test:
	name: Run tests for Python ${{ matrix.python-version }}
	runs-on: ubuntu-latest
	needs:
	- style
	strategy:
	matrix:
	python-version: [ '3.9', '3.11']
	steps:
	- uses: actions/checkout@v1
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v1
	with:
	python-version: ${{ matrix.python-version }}
	- name: Set up test environment
	run: \|
	python -m pip install --upgrade pip
	pip install .
	less requirements.txt \| grep 'pytest\\|chex' \| xargs -i -t pip install {}
	- name: Run tests
	run: \|
	pytest -n auto -vv -m "not benchmark" --cov=blackjax --cov-report=xml --cov-report=term tests
	- name: Upload coverage to Codecov
	uses: codecov/codecov-action@v1
	with:
	env_vars: OS,PYTHON
	name: codecov-umbrella
	fail_ci_if_error: false

Provide feedback