FRML-101 Refactors `train.py` and affected code #52

	name: Python CI

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

	jobs:
	build:

	runs-on: ubuntu-latest
	strategy:
	fail-fast: false
	matrix:
	python-version: [ "3.9", "3.10", "3.11" ]

	steps:
	- uses: actions/checkout@v3

	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v3
	with:
	python-version: ${{ matrix.python-version }}

	- name: Set up gcloud
	id: 'auth'
	uses: 'google-github-actions/auth@v1'
	with:
	credentials_json: '${{ secrets.FRDC_DOWNLOAD_KEY }}'

	- name: 'Set up Cloud SDK'
	uses: 'google-github-actions/setup-gcloud@v1'

	# We don't necessarily need to install the CUDA version of torch, so we'll do with cpu
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	python -m pip install flake8 pytest poetry
	poetry export --with dev --without-hashes -o requirements.txt
	pip install -r requirements.txt
	pip install torch torchaudio torchvision lightning

	- name: Lint with flake8
	run: \|
	# stop the build if there are Python syntax errors or undefined names
	flake8 src/ --count --select=E9,F63,F7,F82 --show-source --statistics
	# exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
	flake8 src/ --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics

	- name: Test with pytest
	run: \|
	pytest

Provide feedback