Merge pull request #6 from goodwin64/patch-1 #7

	name: CI

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

	jobs:

	lint:
	runs-on: ubuntu-latest
	steps:
	- uses: actions/checkout@v2
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v2
	with:
	python-version: '3.10'
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install flake8 black
	- name: Run Flake8
	run: flake8 ./ --config=.flake8
	- name: Run Black
	run: black ./ --config=pyproject.toml --check

	test:
	runs-on: ubuntu-latest
	needs: [lint]
	steps:
	- uses: actions/checkout@v2
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v2
	with:
	python-version: '3.10'
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install albumentations==1.2.1 numpy==1.21.5 rich==13.0.0 hydra-core==1.3.1 matplotlib==3.5.3 pytest torch==1.12.0+cpu torchvision==0.13.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
	- name: Run Tests
	run: pytest

Provide feedback