pyproject.toml

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

[project]
name = "halt-cot"
version = "0.1.0"
description = "Entropy-based early stopping for chain-of-thought reasoning."
readme = "README.md"
requires-python = ">=3.10"
license = { text = "MIT License" }
authors = [{ name = "Yassir Laaouach" }]
keywords = ["chain-of-thought", "entropy", "early-stopping", "reasoning", "transformers"]
classifiers = [
  "Development Status :: 3 - Alpha",
  "Intended Audience :: Science/Research",
  "License :: OSI Approved :: MIT License",
  "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 = []

[project.optional-dependencies]
transformers = [
  "torch>=2.2",
  "transformers>=4.44",
  "accelerate>=0.33",
]
demo = [
  "torch>=2.2",
  "transformers>=4.44",
  "accelerate>=0.33",
  "gradio>=4.44",
  "huggingface_hub>=0.24",
]
publish = [
  "huggingface_hub>=0.24",
]
dev = [
  "pytest>=8",
]

[project.scripts]
halt-cot = "halt_cot.cli:main"

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

[tool.pytest.ini_options]
testpaths = ["tests"]
addopts = "-q"