Internal change #131

	# Copyright 2021 DeepMind Technologies Limited. All Rights Reserved.
	#
	# Licensed under the Apache License, Version 2.0 (the "License");
	# you may not use this file except in compliance with the License.
	# You may obtain a copy of the License at
	#
	# http://www.apache.org/licenses/LICENSE-2.0
	#
	# Unless required by applicable law or agreed to in writing, software
	# distributed under the License is distributed on an "AS IS" BASIS,
	# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	# See the License for the specific language governing permissions and
	# limitations under the License.
	# ============================================================================

	name: pytest

	on: [push, pull_request]

	jobs:
	test-ubuntu:
	runs-on: ubuntu-latest
	strategy:
	matrix:
	python-version: [3.7, 3.8, 3.9]
	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: \|
	pip install --upgrade pip setuptools
	pip install .
	pip install .[testing]
	- name: Check types with pytype
	run: \|
	pytype -j "$(grep -c ^processor /proc/cpuinfo)" enn
	- name: Test with pytest
	run: \|
	pytest -n "$(grep -c ^processor /proc/cpuinfo)" enn

Provide feedback