# automatically generated by g-sorcery
# please do not edit this file

EAPI=8

REALNAME="${PN}"
LITERALNAME="${PN}"
REALVERSION="${PV}"
DIGEST_SOURCES="yes"
PYTHON_COMPAT=( python{3_11,3_12,3_13,3_14} )
DISTUTILS_USE_PEP517=standalone

inherit python-r1 gs-pypi

DESCRIPTION="PyTorch native Metrics"

HOMEPAGE="https://github.com/Lightning-AI/torchmetrics"
LICENSE="Apache-2.0"
SRC_URI="https://files.pythonhosted.org/packages/source/${REALNAME::1}/${REALNAME}/${REALNAME}-${REALVERSION}.tar.gz"
SOURCEFILE="${REALNAME}-${REALVERSION}.tar.gz"
RESTRICT="test"

SLOT="0"
KEYWORDS="~amd64 ~x86"

IUSE="all audio clustering detection image multimodal text typing video visual"
DEPENDENCIES=">dev-python/numpy-1.20.0[${PYTHON_USEDEP}]
	>dev-python/packaging-17.1[${PYTHON_USEDEP}]
	dev-python/torch[${PYTHON_USEDEP}]
	dev-python/lightning-utilities[${PYTHON_USEDEP}]
	audio? ( dev-python/onnxruntime[${PYTHON_USEDEP}] )
	audio? ( >=dev-python/requests-2.19.0[${PYTHON_USEDEP}] )
	audio? ( dev-python/torchaudio[${PYTHON_USEDEP}] )
	audio? ( dev-python/Gammatone[${PYTHON_USEDEP}] )
	audio? ( dev-python/pystoi[${PYTHON_USEDEP}] )
	audio? ( dev-python/pesq[${PYTHON_USEDEP}] )
	audio? ( dev-python/librosa[${PYTHON_USEDEP}] )
	clustering? ( dev-python/torch-linear-assignment[${PYTHON_USEDEP}] )
	detection? ( dev-python/pycocotools[${PYTHON_USEDEP}] )
	detection? ( dev-python/torchvision[${PYTHON_USEDEP}] )
	image? ( dev-python/torch-fidelity[${PYTHON_USEDEP}] )
	image? ( dev-python/torchvision[${PYTHON_USEDEP}] )
	image? ( >dev-python/scipy-1.0.0[${PYTHON_USEDEP}] )
	multimodal? ( dev-python/piq[${PYTHON_USEDEP}] )
	multimodal? ( dev-python/einops[${PYTHON_USEDEP}] )
	multimodal? ( dev-python/transformers[${PYTHON_USEDEP}] )
	multimodal? ( dev-python/timm[${PYTHON_USEDEP}] )
	text? ( dev-python/transformers[${PYTHON_USEDEP}] )
	text? ( >=dev-python/regex-2021.9.24[${PYTHON_USEDEP}] )
	text? ( dev-python/sentencepiece[${PYTHON_USEDEP}] )
	text? ( dev-python/nltk[${PYTHON_USEDEP}] )
	text? ( <dev-python/tqdm-4.68.0[${PYTHON_USEDEP}] )
	text? ( dev-python/mecab-python3[${PYTHON_USEDEP}] )
	text? ( dev-python/ipadic[${PYTHON_USEDEP}] )
	typing? ( ~dev-python/mypy-1.17.1[${PYTHON_USEDEP}] )
	typing? ( dev-python/types-six[${PYTHON_USEDEP}] )
	typing? ( dev-python/torch[${PYTHON_USEDEP}] )
	typing? ( dev-python/types-emoji[${PYTHON_USEDEP}] )
	typing? ( dev-python/types-protobuf[${PYTHON_USEDEP}] )
	typing? ( dev-python/types-setuptools[${PYTHON_USEDEP}] )
	typing? ( dev-python/types-requests[${PYTHON_USEDEP}] )
	typing? ( dev-python/types-tabulate[${PYTHON_USEDEP}] )
	typing? ( dev-python/types-PyYAML[${PYTHON_USEDEP}] )
	video? ( dev-python/einops[${PYTHON_USEDEP}] )
	video? ( dev-python/vmaf-torch[${PYTHON_USEDEP}] )
	visual? ( dev-python/SciencePlots[${PYTHON_USEDEP}] )
	visual? ( >=dev-python/matplotlib-3.6.0[${PYTHON_USEDEP}] )
	all? ( dev-python/onnxruntime[${PYTHON_USEDEP}] )
	all? ( >=dev-python/requests-2.19.0[${PYTHON_USEDEP}] )
	all? ( dev-python/torchaudio[${PYTHON_USEDEP}] )
	all? ( dev-python/Gammatone[${PYTHON_USEDEP}] )
	all? ( dev-python/pystoi[${PYTHON_USEDEP}] )
	all? ( dev-python/pesq[${PYTHON_USEDEP}] )
	all? ( dev-python/librosa[${PYTHON_USEDEP}] )
	all? ( dev-python/torch-linear-assignment[${PYTHON_USEDEP}] )
	all? ( dev-python/pycocotools[${PYTHON_USEDEP}] )"
BDEPEND="${DEPENDENCIES}"
RDEPEND="${DEPENDENCIES}"