[build-system]
requires = ["setuptools>=61"]
build-backend = "setuptools.build_meta"

[project]
name = "flashtrace"
version = "0.1.1"
description = "Efficient multi-token attribution for reasoning language models."
readme = "README.md"
requires-python = ">=3.10"
authors = [
    { name = "Wenbo Pan" },
]
license = "MIT"
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Science/Research",
    "Operating System :: OS Independent",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
    "accelerate>=1.11.0",
    "matplotlib>=3.6",
    "networkx>=3.3",
    "numpy>=2.0",
    "seaborn>=0.13.2",
    "spacy>=3.8",
    "torch>=2.5",
    "tqdm>=4.67",
    "transformers>=4.53",
    "wordfreq>=3.1.1",
]

[project.optional-dependencies]
baselines = [
    "bert-score>=0.3.13",
    "evaluate>=0.4.6",
    "sentence-transformers>=4.1.0",
]
eval = [
    "datasets>=2.21",
    "evaluate>=0.4.6",
]
dev = [
    "pytest>=8.0",
]

[dependency-groups]
dev = [
    "pytest>=8.0",
]

[project.scripts]
flashtrace = "flashtrace.cli:main"

[project.urls]
Homepage = "https://github.com/wbopan/flashtrace"
Repository = "https://github.com/wbopan/flashtrace"
Issues = "https://github.com/wbopan/flashtrace/issues"

[tool.setuptools.packages.find]
include = ["flashtrace*"]

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