Fixed failing :doctest_test #176

	name: pytest

	on:
	pull_request:
	branches:
	- v2
	push:
	branches:
	- v2

	jobs:
	test-ubuntu:
	name: "pytest on ${{ matrix.python-version }} on ${{ matrix.os }}"
	runs-on: "${{ matrix.os }}"
	strategy:
	matrix:
	python-version: [3.8, 3.9, '3.10']
	os: [ubuntu-latest]
	steps:
	- uses: actions/checkout@v2
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v1
	with:
	python-version: ${{ matrix.python-version }}
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install -r requirements.txt
	pip install -r requirements-tf.txt
	pip install -r requirements-test.txt
	pip install .
	- name: Test with pytest
	run: \|
	pip install pytest pytest-xdist
	pytest -n auto sonnet --ignore=sonnet/src/conformance/

Provide feedback