add `predict_proba` functionality #104

	name: Unit testing

	on: push

	jobs:
	unit-testing:
	runs-on: ubuntu-latest
	steps:
	- name: Checkout repo
	uses: actions/checkout@v3

	- name: Setup Python
	uses: actions/setup-python@v4

	- name: Install Poetry
	uses: snok/install-poetry@v1
	with:
	virtualenvs-create: true
	virtualenvs-in-project: true

	# Allow loading a cached venv created in a previous run if the lock file is identical
	- name: Load cached venv if it exists
	id: venv-cache
	uses: actions/cache@v3
	with:
	path: .venv
	key: venv-${{ runner.os }}-${{ hashFiles('/poetry.lock', '/pyproject.toml') }}

	- name: Install dependencies
	run: poetry install --no-interaction --extras all_models

	- name: Run tests
	run: poetry run pytest

Provide feedback