pyproject.toml

[project]
name = "optispeech"
version = "1.0"
description = "A lightweight End-to-End text-to-speech model"
dependencies = [
    "torch==2.5.1",
    "torchvision==0.20.1",
    "lightning>=2.0.0",
    "torchmetrics>=0.11.4",
    "nnaudio>=0.3.3",
    "penn==0.0.14",
    "pyworld>=0.3.4",
    "pesq>=0.0.4",
    "pyloudnorm==0.1.1",
    "hydra-core>=1.3.2",
    "hydra-colorlog==1.2.0",
    "hydra-optuna-sweeper==1.2.0",
    "rootutils>=1.0.7",
    "pre-commit>=3.8.0",
    "rich>=13.7.1",
    "pytest>=8.3.2",
    "tensorboard>=2.17.0",
    "librosa==0.9.2",
    "cython>=3.0.11",
    "einops>=0.8.0",
    "unidecode>=1.3.8",
    "scipy>=1.14.0",
    "torchaudio==2.5.1",
    "matplotlib>=3.9.1.post1",
    "pandas>=2.2.2",
    "notebook>=7.2.1",
    "ipywidgets>=8.1.3",
    "gradio==5.9.1",
    "gdown>=5.2.0",
    "wget>=3.2",
    "seaborn>=0.13.2",
    "transformers>=4.44.0",
    "tqdm>=4.66.5",
    "more-itertools>=10.4.0",
    "onnx>=1.16.2",
    "onnxruntime>=1.18.1",
    "requests>=2.32.3",
    "numpy==1.26",
    "torchcrepe>=0.0.23",
    "piper-phonemize>=1.1.0 ; sys_platform == 'linux'",
]
readme = "README.md"
requires-python = ">= 3.11"

[dependency-groups]
dev = [
    "cython>=3.0.11",
    "black>=24.8.0",
    "isort>=5.13.2",
    "pytest>=8.3.2",
]


[project.scripts]
data-process = 'optispeech.tools.preprocess_dataset:main'
data-stats = 'optispeech.tools.generate_data_statistics:main'
onnx-export = 'optispeech.onnx.export:main'
onnx-infer = 'optispeech.onnx.infer:main'

[tool.uv]
managed = true

[tool.uv.pip]
universal = true

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.metadata]
allow-direct-references = true

[tool.hatch.build.targets.wheel]
packages = ["optispeech"]

[tool.black]
line-length = 120
target-version = ['py310']
exclude = '''
(
  /(
      \.eggs         # exclude a few common directories in the
    | \.git          # root of the project
    | \.hg
    | \.mypy_cache
    | \.tox
    | \.venv
    | _build
    | buck-out
    | build
    | dist
  )/
  | foo.py           # also separately exclude a file named foo.py in
                     # the root of the project
)
'''

[tool.pytest.ini_options]
addopts = [
  "--color=yes",
  "--durations=0",
  "--strict-markers",
  "--doctest-modules",
]
filterwarnings = [
  "ignore::DeprecationWarning",
  "ignore::UserWarning",
]
log_cli = "True"
markers = [
  "slow: slow tests",
]
minversion = "6.0"
testpaths = "tests/"

[tool.coverage.report]
exclude_lines = [
    "pragma: nocover",
    "raise NotImplementedError",
    "raise NotImplementedError()",
    "if __name__ == .__main__.:",
]