Allow specifying datasets in configs, change default optimizer, improved CUDA support #18

	name: CI

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

	jobs:
	quality:
	runs-on: ubuntu-latest
	steps:
	- uses: actions/checkout@v4

	- name: Install uv
	uses: astral-sh/setup-uv@v1
	with:
	enable-cache: true
	cache-dependency-glob: "requirements.txt"

	- name: Set up Python
	uses: actions/setup-python@v5
	with:
	python-version: "3.12"

	- name: Install dependencies
	run: \|
	uv sync

	- name: Check formatting (Black)
	run: uv run black --check .

	- name: Lint (Ruff)
	run: uv run ruff check .

	- name: Type Check (Pyre)
	# Note: This might fail if the codebase has existing type errors.
	# We run it to track progress.
	run: uv run pyre check \|\| echo "Pyre found errors, but we continue for now."

Provide feedback