# Copyright 2025 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2

EAPI=8

DISTUTILS_SINGLE_IMPL=1
PYTHON_COMPAT=( python3_{10..13} )
DISTUTILS_USE_PEP517=setuptools

inherit distutils-r1 pypi

DESCRIPTION="Curated transformer models for spaCy pipelines."
HOMEPAGE="https://github.com/explosion/spacy-curated-transformers"

LICENSE="MIT"
SLOT="0"
KEYWORDS="~amd64"

# shellcheck disable=SC2016
RDEPEND="$(python_gen_cond_dep '
		<dev-python/curated-tokenizers-3.0.0[${PYTHON_USEDEP}]
		>=dev-python/curated-tokenizers-2.0.0[${PYTHON_USEDEP}]
		<dev-python/curated-transformers-3.0.0[${PYTHON_USEDEP}]
		>=dev-python/curated-transformers-2.0.0[${PYTHON_USEDEP}]
		>=dev-python/fsspec-2023.5.0[${PYTHON_USEDEP}]
		<dev-python/thinc-9.2.0[${PYTHON_USEDEP}]
		>=dev-python/thinc-9.0.0_pre4[${PYTHON_USEDEP}]
	')
	>=sci-ml/pytorch-1.12.0[${PYTHON_SINGLE_USEDEP}]"