Introduce graph classes for graph-based models #131

	name: Unit Tests

	on:
	# trigger on pushes to any branch
	push:
	# and also on PRs to main
	pull_request:
	branches:
	- main

	jobs:
	build:
	runs-on: ubuntu-latest
	strategy:
	matrix:
	python-version: ["3.9", "3.10", "3.11", "3.12"]

	steps:
	- uses: actions/checkout@v3
	- 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
	if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
	pip install torch-geometric>=2.5.2
	- name: Load cache data
	uses: actions/cache/restore@v4
	with:
	path: data
	key: ${{ runner.os }}-meps-reduced-example-data-v0.1.0
	restore-keys: \|
	${{ runner.os }}-meps-reduced-example-data-v0.1.0
	- name: Test with pytest
	run: \|
	pytest -v -s
	- name: Save cache data
	uses: actions/cache/save@v4
	with:
	path: data
	key: ${{ runner.os }}-meps-reduced-example-data-v0.1.0

Provide feedback