# 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="Catalyst. Accelerated deep learning R&D with PyTorch."

HOMEPAGE="https://github.com/catalyst-team/catalyst"
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 comet cv deepspeed dev ml mlflow neptune onnx onnx-gpu optuna profiler wandb"
DEPENDENCIES=">=dev-python/numpy-1.18[${PYTHON_USEDEP}]
	dev-python/torch[${PYTHON_USEDEP}]
	dev-python/accelerate[${PYTHON_USEDEP}]
	dev-python/hydra-slayer[${PYTHON_USEDEP}]
	>=dev-python/tqdm-4.33.0[${PYTHON_USEDEP}]
	dev-python/tensorboardX[${PYTHON_USEDEP}]
	all? ( >=dev-python/imageio-2.5.0[${PYTHON_USEDEP}] )
	all? ( dev-python/opencv-python-headless[${PYTHON_USEDEP}] )
	all? ( <dev-python/scikit-image-0.19.0[${PYTHON_USEDEP}] )
	all? ( dev-python/torchvision[${PYTHON_USEDEP}] )
	all? ( >=dev-python/pillow-6.1[${PYTHON_USEDEP}] )
	all? ( dev-python/requests[${PYTHON_USEDEP}] )
	all? ( >=dev-python/scipy-1.4.1[${PYTHON_USEDEP}] )
	all? ( >=dev-python/matplotlib-3.1.0[${PYTHON_USEDEP}] )
	all? ( >=dev-python/pandas-1.0.0[${PYTHON_USEDEP}] )
	all? ( >=dev-python/scikit-learn-1.0[${PYTHON_USEDEP}] )
	all? ( dev-python/optuna[${PYTHON_USEDEP}] )
	comet? ( dev-python/comet-ml[${PYTHON_USEDEP}] )
	cv? ( >=dev-python/imageio-2.5.0[${PYTHON_USEDEP}] )
	cv? ( dev-python/opencv-python-headless[${PYTHON_USEDEP}] )
	cv? ( <dev-python/scikit-image-0.19.0[${PYTHON_USEDEP}] )
	cv? ( dev-python/torchvision[${PYTHON_USEDEP}] )
	cv? ( >=dev-python/pillow-6.1[${PYTHON_USEDEP}] )
	cv? ( dev-python/requests[${PYTHON_USEDEP}] )
	deepspeed? ( dev-python/deepspeed[${PYTHON_USEDEP}] )
	dev? ( dev-python/pytest[${PYTHON_USEDEP}] )
	dev? ( ~dev-python/sphinx-2.2.1[${PYTHON_USEDEP}] )
	dev? ( <=dev-python/jinja2-3.0.3[${PYTHON_USEDEP}] )
	dev? ( ~dev-python/docutils-0.17.1[${PYTHON_USEDEP}] )
	dev? ( ~dev-python/mock-3.0.5[${PYTHON_USEDEP}] )
	dev? ( dev-python/catalyst-codestyle[${PYTHON_USEDEP}] )
	dev? ( ~dev-python/black-21.8_beta0[${PYTHON_USEDEP}] )
	dev? ( <=dev-python/click-8.0.4[${PYTHON_USEDEP}] )
	dev? ( dev-python/catalyst-sphinx-theme[${PYTHON_USEDEP}] )
	dev? ( ~dev-python/tomlkit-0.7.2[${PYTHON_USEDEP}] )
	dev? ( dev-python/pre-commit[${PYTHON_USEDEP}] )
	dev? ( dev-python/path[${PYTHON_USEDEP}] )
	ml? ( >=dev-python/scipy-1.4.1[${PYTHON_USEDEP}] )
	ml? ( >=dev-python/matplotlib-3.1.0[${PYTHON_USEDEP}] )
	ml? ( >=dev-python/pandas-1.0.0[${PYTHON_USEDEP}] )
	ml? ( >=dev-python/scikit-learn-1.0[${PYTHON_USEDEP}] )
	mlflow? ( dev-python/mlflow[${PYTHON_USEDEP}] )
	neptune? ( dev-python/neptune-client[${PYTHON_USEDEP}] )
	onnx? ( dev-python/onnx[${PYTHON_USEDEP}] )
	onnx? ( dev-python/onnxruntime[${PYTHON_USEDEP}] )
	onnx-gpu? ( dev-python/onnx[${PYTHON_USEDEP}] )
	onnx-gpu? ( dev-python/onnxruntime-gpu[${PYTHON_USEDEP}] )
	optuna? ( dev-python/optuna[${PYTHON_USEDEP}] )
	profiler? ( dev-python/torch-tb-profiler[${PYTHON_USEDEP}] )
	wandb? ( dev-python/wandb[${PYTHON_USEDEP}] )"
BDEPEND="${DEPENDENCIES}"
RDEPEND="${DEPENDENCIES}"