code/TaoTrain/pyproject.toml

[build-system]
requires = ["setuptools>=68.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "taoTrain"
version = "0.1.0"
description = "Clean, modular PyTorch LLM training framework with pluggable architectures, AimStack logging, and TUI inference"
readme = "README.md"
requires-python = ">=3.10"
license = { text = "MIT" }
authors = [
    { name = "Felix", email = "felix@example.com" }
]

dependencies = [
    "torch>=2.0.0",
    "transformers>=4.30.0",
    "datasets>=2.10.0",
    "pydantic>=2.0.0",
    "pydantic-settings>=2.0.0",
    "aim>=3.15.0",
    "click>=8.1.0",
    "rich>=13.0.0",
    "textual>=0.30.0",
    "numpy>=1.24.0",
    "tqdm>=4.65.0",
    "sentencepiece>=0.1.99",
]

[project.optional-dependencies]
dev = [
    "pytest>=7.4.0",
    "pytest-cov>=4.1.0",
    "pytest-xdist>=3.3.0",
    "black>=23.7.0",
    "ruff>=0.0.280",
    "typing-extensions>=4.7.0",
]

[project.scripts]
train = "taoTrain.cli:main"
train-tokenizer = "taoTrain.cli:train_tokenizer_command"
tui-chat = "taoTrain.inference.tui:main"

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

[tool.setuptools.package-data]
taoTrain = ["configs/**/*.yaml"]

[tool.black]
line-length = 100
target-version = ["py310"]

[tool.ruff]
line-length = 100
target-version = "py310"
select = ["E", "F", "W", "I", "N", "UP", "RUF"]
ignore = ["E501"]

[tool.pytest.ini_options]
testpaths = ["tests"]
python_files = "test_*.py"
addopts = "--verbose"