[project]
name = "sonicembed-eval"
version = "0.1.0"
description = "Song Describer, CLAP and MTEB evaluation for SonicEmbed"
requires-python = ">=3.11,<3.13"
dependencies = [
    "torch>=2.8",
    "transformers>=5.20",
    "sentence-transformers>=6.1",
    "huggingface-hub>=1.0",
    "librosa>=0.11",
    "soundfile>=0.13",
    "pyarrow>=20",
    "numpy>=2",
    "tqdm",
    "mteb>=2.26",
    "pillow>=11",
    "torchvision>=0.23",
    "laion-clap>=1.1.7",
]

# laion-clap's own numpy pin is older than the rest of these need.
[tool.uv]
override-dependencies = ["numpy>=2"]

# PyPI's torch is CPU-only on Windows; Linux wheels already include CUDA.
[tool.uv.sources]
torch = { index = "pytorch-cu128", marker = "sys_platform == 'win32'" }
torchvision = { index = "pytorch-cu128", marker = "sys_platform == 'win32'" }

[[tool.uv.index]]
name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu128"
explicit = true