Add type annotation to func args and return type #4

	name: Ubuntu

	on: [push, pull_request]

	jobs:
	cpu:
	runs-on: ubuntu-latest
	if: "!contains(github.event.head_commit.message, 'ci skip')"
	strategy:
	matrix:
	python-version: ["3.11"]
	steps:
	- name: Cancel previous run
	uses: styfle/[email protected]
	with:
	access_token: ${{ github.token }}
	- uses: actions/checkout@v3
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v4
	with:
	python-version: ${{ matrix.python-version }}
	# use poetry and cache installed packages, see https://github.com/marketplace/actions/python-poetry-action
	- name: Install poetry
	uses: abatilo/actions-poetry@v2
	- name: Setup a local virtual environment (if no poetry.toml file)
	run: \|
	poetry config virtualenvs.create true --local
	poetry config virtualenvs.in-project true --local
	- uses: actions/cache@v3
	name: Define a cache for the virtual environment based on the dependencies lock file
	with:
	path: ./.venv
	key: venv-${{ hashFiles('poetry.lock') }}
	- name: Install the project dependencies
	run: \|
	poetry install --with dev --extras "envpool"
	- name: wandb login
	run: \|
	poetry run wandb login e2366d661b89f2bee877c40bee15502d67b7abef
	- name: Test with pytest
	# ignore test/throughput which only profiles the code
	run: \|
	poetry run poe test
	- name: Upload coverage to Codecov
	uses: codecov/codecov-action@v1
	with:
	token: ${{ secrets.CODECOV }}
	file: ./coverage.xml
	flags: unittests
	name: codecov-umbrella
	fail_ci_if_error: false

Provide feedback