pyproject.toml

[project]
name = "trust-the-guard"
version = "0.1.0"
description = "An activation-steering game where players persuade a Gemma 4 guard."
readme = "README.md"
requires-python = ">=3.11"
dependencies = [
    "gradio==5.49.1",
    "huggingface-hub>=1.23.0",
    "pillow>=11.3.0,<12",
    "python-dotenv>=1.2.2",
    "safetensors>=0.8.0",
]

[project.optional-dependencies]
gpu = [
    "accelerate>=1.14.0",
    "scikit-learn>=1.9.0",
    "sentencepiece>=0.2.2",
    "torch==2.11.0",
    "torchvision==0.26.0",
    "transformers>=5.5.0",
]

[tool.uv]
# The published steering-vectors wheel still declares Transformers <5, but the
# Gemma 4 reference pipeline explicitly requires Transformers >=5.5 and installs
# steering-vectors without dependency metadata. Keep the override visible here.
override-dependencies = [
    "transformers>=5.5.0",
    "huggingface-hub>=1.0",
]

[tool.ruff]
line-length = 100

[tool.pytest.ini_options]
testpaths = ["tests"]
pythonpath = ["."]

[dependency-groups]
dev = [
    "pytest>=9.1.1",
    "ruff>=0.15.21",
]