pyproject.toml

[build-system]
requires = ["setuptools>=68.0", "wheel"]
build-backend = "setuptools.backends.legacy:build"

[project]
name = "neollm"
version = "0.1.0"
description = "A production-grade LLM built from scratch — GPT/LLaMA-style Transformer"
readme = "README.md"
requires-python = ">=3.10"
license = { text = "MIT" }
authors = [{ name = "NeoLLM Project" }]

dependencies = [
    "torch>=2.3.0",
    "tokenizers>=0.19.0",
    "datasets>=2.19.0",
    "huggingface-hub>=0.23.0",
    "numpy>=1.26.0",
    "einops>=0.8.0",
    "tqdm>=4.66.0",
    "pyyaml>=6.0.1",
    "omegaconf>=2.3.0",
    "tensorboard>=2.17.0",
    "rich>=13.7.0",
    "typing_extensions>=4.12.0",
]

[project.optional-dependencies]
dev = ["pytest>=8.2.0", "pytest-cov>=5.0.0"]
flash = ["flash-attn>=2.5.0"]
tracking = ["wandb>=0.17.0"]

[project.scripts]
neollm-train = "scripts.train:main"
neollm-chat = "scripts.chat:main"

[tool.setuptools.packages.find]
where = ["."]
include = ["neollm*", "scripts*"]

[tool.pytest.ini_options]
testpaths = ["tests"]
python_files = ["test_*.py"]
python_classes = ["Test*"]
python_functions = ["test_*"]
addopts = "-v --tb=short"