pyproject.toml

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

[project]
name = "wham"
version = "0.0.1"
description = "Towards A Translative Model of Sperm Whale Vocalization"
readme = "README.md"
authors = [
    { name = "Project CETI" }
]
license = { text = "MIT" }
requires-python = ">=3.10,<3.11"
dependencies = [
    "torch",
    "gradio",
    "argbind>=0.3.2",
    "numpy<1.24",
    "pydantic>=2.0,<3",
    "huggingface_hub",
    "loralib",
    "torch_pitch_shift",
    "soundfile",
    "pydub",
    "tqdm",
    "Cython",
    "pandas",
    "pathlib",
    "ffmpeg-python",
    "scikit-learn",
    "wandb",
    "gdown",
    "transformers",
    "fadtk",
    "urllib3>=2.0.2",
    "plotly",
    "pyharp",
    "ruff",
    "wavebeat @ git+https://github.com/hugofloresgarcia/wavebeat.git",
    "lac @ git+https://github.com/hugofloresgarcia/lac.git",
    "descript-audiotools @ git+https://github.com/hugofloresgarcia/audiotools.git"
]

[tool.setuptools.packages.find]
where = ["."]
include = ["wham*", "vampnet*"]

[tool.ruff]
target-version = "py310"
line-length = 88

[tool.ruff.lint]
select = ["E", "F", "W", "I"]
fixable = ["ALL"]

[tool.ruff.lint.isort]
known-first-party = ["wham", "vampnet"]
section-order = ["future", "standard-library", "third-party", "first-party", "local-folder"]