Fix `checkpoint_model` #378

	name: Tests

	on:
	push:
	paths: ["*/.py", .github/workflows/test.yml]
	branches: [main]
	pull_request:
	paths: ["*/.py", .github/workflows/test.yml]
	branches: [main]

	jobs:
	tests:
	runs-on: ubuntu-latest
	steps:
	- name: Check out repo
	uses: actions/checkout@v4

	- name: Set up Python
	uses: actions/setup-python@v5
	with:
	python-version: 3.9
	cache: pip
	cache-dependency-path: pyproject.toml

	- name: Install uv
	run: pip install uv

	- name: Install dependencies
	run: \|
	pip install torch==2.2.1 --index-url https://download.pytorch.org/whl/cpu
	pip install torch-scatter -f https://data.pyg.org/whl/torch-2.2.1+cpu.html
	uv pip install .[test] --system

	- name: Run Tests
	run: pytest --capture=no --cov .

Provide feedback