pyproject.toml

[tool.poetry]
name = "shepardtts"
version = "0.0.1-dev"
description = "ShepardTTS is a fine-tuned XTTS model, trained on paired dialogue/audio samples from the Mass Effect 2 and Mass Effect 3 base games."
authors = ["Patrick Darwinkel"]
readme = "README.md"
packages = [{include = "shepardtts"}]

[[tool.poetry.source]]
name = "torch-cpu"
url = "https://download.pytorch.org/whl/cpu"
priority = "explicit"

[[tool.poetry.source]]
name = "torch-intel"
url = "https://pytorch-extension.intel.com/release-whl/stable/cpu/us/"
priority = "explicit"

[tool.poetry-dynamic-versioning]
enable = true
vcs = "git"
style = "pep440"
metadata = true
tagged-metadata = true
dirty = true
bump = true

# Build configuration
[build-system]
requires = ["poetry-core>=1.0.0", "poetry-dynamic-versioning>=1.0.0,<2.0.0"]
build-backend = "poetry_dynamic_versioning.backend"

# Poetry dependency configuration
[tool.poetry.dependencies]
python = "^3.11,<3.13"
pandas = "1.5.3"
numpy = "1.26.4"
coqui-tts = "0.23.1" # Py3.12 fork
clean-text = "0.6.0"
num2words = "0.5.13"
unidecode = "1.3.8"

[tool.poetry.group.gpu.dependencies]
torch = "2.2.2"
torchaudio = "2.2.2"
deepspeed = "0.14.2"

[tool.poetry.group.cpu.dependencies]
torch = {version = "^2.2.2+cpu", source = "torch-cpu"}
torchaudio = {version = "^2.2.2+cpu", source = "torch-cpu"}
intel-extension-for-pytorch  = {version = "^2.2.0+cpu", source = "torch-intel"}
oneccl_bind_pt  = {version = "^2.2.0+cpu", source = "torch-intel"}

[tool.poetry.group.dev.dependencies]
ruff = "0.4.4"
mypy = "1.9.0"
poetry-plugin-export = "1.7.1"
pytest = "8.1.1"
pytest-cov = "5.0.0"

[tool.poetry.group.train.dependencies]
coqui-tts-trainer = "0.1.0" # Py3.12 fork
openpyxl = "3.1.2"
datasets = "2.18.0"
soundfile = "0.12.1"
accelerate = "0.30.1"

[tool.poetry.group.deploy.dependencies]
gradio = "4.26.0"

# Code quality tooling configuration
[tool.ruff]
line-length = 120
target-version = "py312"
fix = true

[tool.ruff.lint]
select = ["ALL"]
fixable = ["ALL"]
ignore = ["D203", "D212", "COM812", "ISC001", "T201"]

[tool.ruff.lint.per-file-ignores]
"app.py" = ["PLR0913"] # Not relevant for Gradio
"train.py" = ["PTH"] # Fix some other time
"test_*.py" = ["S101"] # Allow assert in tests

[tool.mypy]
python_version = "3.12"
strict = true
ignore_missing_imports = true