[project]
name = "lowbit"
version = "0.1.0"
description = "Marco-Nano-T ternary distillation trials"
requires-python = ">=3.12,<3.13"
dependencies = [
    "torch",
    "transformers",
    "datasets",
    "safetensors",
    "numpy",
    "zstandard",
    "pyyaml",
    "pyarrow",
    "lm-eval[ifeval,math]",  # IFEval and MATH-500 scoring for RS-full
    "accelerate",  # lm-eval's HF backend (T0.3 eval wrapper)
    "pytest",
    "hypothesis",
    "fasttext-numpy2-wheel",  # language ID (lid.176) for prompt translation and output filters (T1.3, T1.4)
]

[project.optional-dependencies]
# gen: desktop GPU (sm_120), x86-64 — teacher generation and student evals
gen = ["vllm"]
# spark: GB10, aarch64, CUDA 13 — resolved on spark itself (env/spark.lock)
spark = ["torchao", "bitsandbytes"]  # 8-bit AdamW: whichever installs on aarch64 and passes T2.5

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

[tool.setuptools.packages.find]
where = ["src"]

[tool.pytest.ini_options]
testpaths = ["tests"]
markers = ["gpu: needs a CUDA device (run with -m gpu)",
           "slow: reads full datasets from $MT_DATA (run with -m slow)"]
addopts = "-m 'not gpu and not slow'"