Fix for joblib not saturating CPU during multiprocessing #354

	name: Continuous Integration

	on: [pull_request, workflow_dispatch]

	defaults:
	run:
	shell: bash

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

	jobs:
	ci:
	name: CI
	runs-on: ubuntu-latest
	if: \|
	github.event_name == 'push' \|\|
	(github.event_name == 'pull_request' && github.event.pull_request.draft == false)
	strategy:
	fail-fast: false
	matrix:
	python-version: ['3.8']
	torch-version: ['1.11', '2.2']

	steps:
	- name: Checkout
	uses: actions/checkout@v4
	with:
	fetch-depth: 0
	lfs: false

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

	- name: Cache dependency
	id: cache-dependencies
	uses: actions/cache@v4
	with:
	path: ~/.cache/pip
	key: ${{ runner.os }}-pip-${{ matrix.python-version }}-${{ matrix.torch-version }}$
	restore-keys: \|
	${{ runner.os }}-pip-${{ matrix.python-version }}-
	${{ runner.os }}-pip-

	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install torch==${{ matrix.torch-version }} --extra-index-url https://download.pytorch.org/whl/cpu
	pip install -e .[dev]

	- name: Run the fast CPU tests with coverage
	run: \|
	pytest -v -x -n auto -m "not slow and not gpu" --cov=sbi --cov-report=xml tests/

	- name: Upload coverage to Codecov
	uses: codecov/codecov-action@v4
	with:
	env_vars: OS,PYTHON
	file: ./coverage.xml
	flags: unittests
	name: codecov-sbi-fast-cpu
	token: ${{ secrets.CODECOV_TOKEN }}

Provide feedback