pyproject.toml

[project]
name = "trm"
version = "0.1.0"
description = "Tiny Recursive Model (TRM) implementation for Sudoku-Extreme"
readme = "README.md"
requires-python = "==3.12.7"
dependencies = [
    "torch==2.8.0",
    "numpy>=1.26.0",
    "tqdm>=4.65.0",
    "argdantic>=1.3.3",
    "huggingface-hub>=1.0.1",
    "trackio>=0.25.0",
    "requests>=2.32.5",
]

[project.optional-dependencies]
dev = [
    "pytest>=7.0.0",
]

[project.urls]
"Paper" = "https://arxiv.org/abs/2510.04871v1"

[tool.hatch.build.targets.wheel]
packages = ["."]

[tool.pytest.ini_options]
testpaths = ["tests"]
python_files = ["test_*.py"]
python_classes = ["Test*"]
python_functions = ["test_*"]
addopts = "-v --tb=short"
markers = [
    "integration: marks tests as integration tests (deselect with '-m \"not integration\"')",
]

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"