Typing in tests: added asserts and cast to remove some mypy errors #30

	name: Ubuntu GPU

	on: [push, pull_request]

	jobs:
	gpu:
	runs-on: [self-hosted, Linux, X64]
	if: "!contains(github.event.head_commit.message, 'ci skip')"
	steps:
	- name: Cancel previous run
	uses: styfle/[email protected]
	with:
	access_token: ${{ github.token }}
	- uses: actions/checkout@v3
	- name: Set up Python 3.11
	uses: actions/setup-python@v4
	with:
	python-version: "3.11"
	# 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
	run: \|
	poetry run poe test

Provide feedback