Small changes to support long, preemptable training runs #669

	name: CI

	on:
	pull_request:
	branches: [master]

	jobs:
	build:

	runs-on: ${{ matrix.os }}
	strategy:
	matrix:
	os: [ubuntu-20.04]
	python-version: ["3.11", "3.10", "3.9"]

	steps:
	- uses: actions/checkout@v3
	with: # no need for the history
	fetch-depth: 1
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v4
	with:
	python-version: ${{ matrix.python-version }}

	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip setuptools
	pip install --upgrade --force-reinstall --no-cache-dir --editable=".[dev]"
	- name: Test with pytest
	run: \|
	pytest
	image-build:
	runs-on: ${{ matrix.os }}
	strategy:
	matrix:
	os: [ubuntu-20.04]
	python-version: ["3.11", "3.10", "3.9"]
	steps:
	- uses: actions/checkout@v3
	with:
	fetch-depth: 1
	- name: Test CPU Docker image build
	run: \|
	docker build -t neuronets/nobrainer:master-cpu -f docker/cpu.Dockerfile .
	- name: Test GPU Docker image build
	run: \|
	docker build -t neuronets/nobrainer:master-gpu -f docker/gpu.Dockerfile .

Provide feedback