File size: 1,546 Bytes
55b60a8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 | [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 = ["."]
|