# Copyright 2024-2026 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2
EAPI=8
DISTUTILS_USE_PEP517=setuptools
DISTUTILS_SINGLE_IMPL=1
PYTHON_COMPAT=( python3_{12..14} )
inherit distutils-r1 pypi
DESCRIPTION="Universal library for loading pretrained PyTorch models"
HOMEPAGE="https://github.com/chaiNNer-org/spandrel https://pypi.org/project/spandrel/"
LICENSE="MIT"
SLOT="0"
KEYWORDS="~amd64"
RDEPEND="
	|| (
		sci-ml/pytorch[${PYTHON_SINGLE_USEDEP}]
		sci-ml/pytorch-rocm[${PYTHON_SINGLE_USEDEP}]
		sci-ml/pytorch-cuda[${PYTHON_SINGLE_USEDEP}]
	)
	$(python_gen_cond_dep 'sci-ml/safetensors[${PYTHON_USEDEP}]')
"