# 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="Train transformer language models with reinforcement learning."

HOMEPAGE="https://github.com/huggingface/trl"
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="benchmark deepspeed dev diffusers llm-judge peft quantization test"
DEPENDENCIES="dev-python/torch[${PYTHON_USEDEP}]
	dev-python/transformers[${PYTHON_USEDEP}]
	<dev-python/numpy-2.0.0[${PYTHON_USEDEP}]
	dev-python/accelerate[${PYTHON_USEDEP}]
	dev-python/datasets[${PYTHON_USEDEP}]
	dev-python/tyro[${PYTHON_USEDEP}]
	dev-python/peft[${PYTHON_USEDEP}]
	dev-python/wandb[${PYTHON_USEDEP}]
	dev-python/deepspeed[${PYTHON_USEDEP}]
	dev-python/stanza[${PYTHON_USEDEP}]
	dev-python/nltk[${PYTHON_USEDEP}]
	dev-python/scipy[${PYTHON_USEDEP}]
	benchmark? ( dev-python/wandb[${PYTHON_USEDEP}] )
	benchmark? ( dev-python/ghapi[${PYTHON_USEDEP}] )
	benchmark? ( dev-python/openrlbenchmark[${PYTHON_USEDEP}] )
	benchmark? ( dev-python/requests[${PYTHON_USEDEP}] )
	benchmark? ( dev-python/deepspeed[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/deepspeed[${PYTHON_USEDEP}] )
	dev? ( dev-python/parameterized[${PYTHON_USEDEP}] )
	dev? ( dev-python/peft[${PYTHON_USEDEP}] )
	dev? ( dev-python/pytest[${PYTHON_USEDEP}] )
	dev? ( dev-python/pytest-xdist[${PYTHON_USEDEP}] )
	dev? ( dev-python/pytest-cov[${PYTHON_USEDEP}] )
	dev? ( dev-python/scikit-learn[${PYTHON_USEDEP}] )
	dev? ( dev-python/pillow[${PYTHON_USEDEP}] )
	dev? ( dev-python/diffusers[${PYTHON_USEDEP}] )
	dev? ( dev-python/deepspeed[${PYTHON_USEDEP}] )
	dev? ( dev-python/wandb[${PYTHON_USEDEP}] )
	dev? ( dev-python/ghapi[${PYTHON_USEDEP}] )
	dev? ( dev-python/openrlbenchmark[${PYTHON_USEDEP}] )
	dev? ( dev-python/requests[${PYTHON_USEDEP}] )
	dev? ( dev-python/deepspeed[${PYTHON_USEDEP}] )
	dev? ( dev-python/bitsandbytes[${PYTHON_USEDEP}] )
	dev? ( dev-python/openai[${PYTHON_USEDEP}] )
	dev? ( dev-python/huggingface-hub[${PYTHON_USEDEP}] )
	dev? ( dev-python/llm-blender[${PYTHON_USEDEP}] )
	diffusers? ( dev-python/diffusers[${PYTHON_USEDEP}] )
	llm-judge? ( dev-python/openai[${PYTHON_USEDEP}] )
	llm-judge? ( dev-python/huggingface-hub[${PYTHON_USEDEP}] )
	llm-judge? ( dev-python/llm-blender[${PYTHON_USEDEP}] )
	peft? ( dev-python/peft[${PYTHON_USEDEP}] )
	quantization? ( dev-python/bitsandbytes[${PYTHON_USEDEP}] )
	test? ( dev-python/parameterized[${PYTHON_USEDEP}] )
	test? ( dev-python/peft[${PYTHON_USEDEP}] )
	test? ( dev-python/pytest[${PYTHON_USEDEP}] )
	test? ( dev-python/pytest-xdist[${PYTHON_USEDEP}] )
	test? ( dev-python/pytest-cov[${PYTHON_USEDEP}] )
	test? ( dev-python/scikit-learn[${PYTHON_USEDEP}] )
	test? ( dev-python/pillow[${PYTHON_USEDEP}] )"
BDEPEND="${DEPENDENCIES}"
RDEPEND="${DEPENDENCIES}"