Model partitioning #106

	name: Tests

	on:
	push:
	branches: [ main ]
	pull_request:
	branches: [ main ]

	jobs:
	tests:
	runs-on: ubuntu-latest

	steps:
	- uses: actions/checkout@v2

	- name: Set up Python 3.9
	uses: actions/setup-python@v2
	with:
	python-version: 3.9

	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install -e .
	pip3 install torch torchvision torchaudio
	pip install pytest
	pip install pytest-cov

	- name: Print PyTorch and CUDA info
	run: \|
	python -c "import torch; print(torch.__version__)"
	python -c "import torch; print(torch.cuda.is_available())"

	- name: Run tests
	run: pytest --color=yes --durations=0 --cov=pipegoose --verbose tests/

	- name: Upload coverage reports to Codecov
	uses: codecov/codecov-action@v3
	env:
	CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}

Provide feedback