pyproject.toml

[project]
name = "torchfx"
version = "0.5.2"
description = "A GPU accelerated library for audio DSP based on PyTorch"
readme = "README.md"
license = { file = "LICENSE" }
authors = [
    { name = "Matteo Spanio", email = "spanio@dei.unipd.it" }
]
requires-python = ">=3.10"
dependencies = [
    "annotated-types>=0.7.0",
    "numpy>=2.2.4",
    "scipy>=1.15.2",
    "setuptools",
    "soundfile>=0.13.1",
    "torch>=2.6.0",
]
keywords = [
    "torch",
    "torchfx",
    "audio",
    "dsp",
    "gpu",
    "accelerated",
    "signal processing",
    "performance",
]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Environment :: GPU",
    "Intended Audience :: Developers",
    "Intended Audience :: Science/Research",
    "Intended Audience :: Telecommunications Industry",
    "License :: OSI Approved :: GNU General Public License v3 (GPLv3)",
    "Natural Language :: English",
    "Topic :: Scientific/Engineering",
    "Topic :: Multimedia :: Sound/Audio",
    "Typing :: Typed",
]

[project.scripts]
torchfx = "cli.__main__:main"

[project.urls]
Repository = "https://github.com/matteospanio/torchfx"
Documentation = "https://matteospanio.github.io/torchfx/"
Changelog = "https://github.com/matteospanio/torchfx/blob/master/CHANGELOG"

[tool.mypy]
python_version = "3.10"
strict = true
check_untyped_defs = true
disallow_untyped_calls = true
disallow_untyped_defs = true
disallow_incomplete_defs = true
warn_unused_ignores = true
warn_return_any = true
warn_redundant_casts = true
warn_unused_configs = true
no_implicit_optional = true
namespace_packages = true
explicit_package_bases = true
mypy_path = "src"

[[tool.mypy.overrides]]
module = [
  "torchfx.*",
  "cli.*",
  "torchaudio.*",
]
ignore_missing_imports = true

[[tool.mypy.overrides]]
module = "benchmarks.*"
ignore_errors = true

[tool.ruff]
src = ["src"]
line-length = 100
target-version = "py310"
lint.select = ["E", "F", "I", "N", "UP", "B", "A", "C4", "SIM", "TID", "ARG"]
lint.ignore = ["E501", "N803", "N806", "N812"]

[tool.ruff.format]
quote-style = "double"
indent-style = "space"
line-ending = "lf"

[tool.black]
line-length = 100
target-version = ["py310"]
skip-string-normalization = false
include = '\.pyi?$'
exclude = '''
/(
    \.git
  | \.mypy_cache
  | \.venv
  | build
  | dist
)/
'''

[tool.coverage.run]
source = ["src/torchfx"]
branch = true

[tool.coverage.report]
# Phase 1 gate: current local CPU-only coverage (with sounddevice omitted)
# is 88%. Ratchet this up as further phases land; never drop below without
# review.
fail_under = 87
# Skip files that are unreachable on a CPU-only workstation. Cluster runs
# exercise these via the sounddevice backend and CUDA kernels.
omit = [
    "src/torchfx/realtime/sounddevice_backend.py",
]

[tool.pytest.ini_options]
minversion = "7.0"
addopts = "--strict-markers --tb=short --benchmark-disable"
testpaths = ["tests"]
pythonpath = ["src"]
markers = [
    "benchmark: performance benchmark (enable with --benchmark-enable)",
]

[tool.docformatter]
recursive = true
blank = true

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

[dependency-groups]
cli = [
    "prompt-toolkit>=3.0",
    "rich>=13.0",
    "tomli>=2.0; python_version < '3.11'",
    "typer>=0.16.0",
    "watchdog>=4.0",
]
dev = [
    "black>=25.1.0",
    "coverage>=7.9.1",
    "mypy>=1.16.1",
    "numba>=0.64.0",
    "pytest>=8.3.5",
    "pytest-benchmark>=5.1.0",
    "ruff>=0.12.1",
    "scalene>=1.5.51",
    "scipy-stubs>=1.15.3.0",
]
realtime = [
    "sounddevice>=0.4.6",
]
cuda = [
    "ninja>=1.11",
]
docs = [
    "ablog>=0.11.12",
    "myst-parser>=4.0.1",
    "pydata-sphinx-theme>=0.16.1",
    "sphinx>=8.1.3",
    "sphinx-design>=0.6.1",
    "sphinxcontrib-bibtex>=2.6.5",
    "sphinxcontrib-mermaid>=1.2.3",
]