ci: switch to amazon runners #16

	name: Unit Tests

	on:
	push:
	branches:
	- main
	pull_request:

	jobs:
	unittest:
	strategy:
	matrix:
	os:
	- "linux.2xlarge"
	- "linux.4xlarge.nvidia.gpu"
	runs-on: ${{ matrix.os }}
	steps:
	# This hash has been reviewed for security issues. If upgrading please
	# rereview the action source.
	- name: Setup Python
	uses: kishaningithub/setup-python-amazon-linux@46d057bc1c866dd51b4d2ad0c7631e9e8eed13e1
	with:
	python-version: '3.10'
	- name: Checkout
	uses: actions/checkout@v2
	- name: Install Dependencies
	run: \|
	set -eux

	sudo yum install -y protobuf-compiler

	pip install -e .[dev] -v
	- name: Run Python Tests
	run: \|
	set -eux

	pytest -v
	- name: Run Rust Lint
	run: \|
	set -eux

	cargo test -v

Provide feedback