# 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="Decision AI Engine"

HOMEPAGE="https://github.com/opendilab/DI-engine"
LICENSE="Apache-2.0"
SRC_URI="https://files.pythonhosted.org/packages/9d/a6/9d1e3852fdb5b06b92302fde50eeed03bc0290e78651036f33890f7a303b/di_engine-${REALVERSION}.tar.gz"
SOURCEFILE="di_engine-${REALVERSION}.tar.gz"
RESTRICT="test"

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

IUSE="bsuite-env common-env dist envpool fast gfootball-env k8s mario minigrid-env procgen-env slimevolleygym-env smac-env sokoban style"
DEPENDENCIES="<=dev-python/setuptools-66.1.1[${PYTHON_USEDEP}]
	~dev-python/yapf-0.29.0[${PYTHON_USEDEP}]
	dev-python/gym[${PYTHON_USEDEP}]
	dev-python/gymnasium[${PYTHON_USEDEP}]
	dev-python/torch[${PYTHON_USEDEP}]
	<dev-python/numpy-2.0[${PYTHON_USEDEP}]
	dev-python/DI-treetensor[${PYTHON_USEDEP}]
	dev-python/DI-toolkit[${PYTHON_USEDEP}]
	dev-python/trueskill[${PYTHON_USEDEP}]
	dev-python/tensorboardX[${PYTHON_USEDEP}]
	dev-python/wandb[${PYTHON_USEDEP}]
	dev-python/matplotlib[${PYTHON_USEDEP}]
	dev-python/easydict[${PYTHON_USEDEP}]
	dev-python/pyyaml[${PYTHON_USEDEP}]
	dev-python/enum-tools[${PYTHON_USEDEP}]
	dev-python/cloudpickle[${PYTHON_USEDEP}]
	dev-python/hickle[${PYTHON_USEDEP}]
	dev-python/tabulate[${PYTHON_USEDEP}]
	>=dev-python/click-7.0.0[${PYTHON_USEDEP}]
	<=dev-python/flask-2.0.3[${PYTHON_USEDEP}]
	<=dev-python/werkzeug-2.0.3[${PYTHON_USEDEP}]
	dev-python/requests[${PYTHON_USEDEP}]
	dev-python/responses[${PYTHON_USEDEP}]
	dev-python/URLObject[${PYTHON_USEDEP}]
	dev-python/pynng[${PYTHON_USEDEP}]
	dev-python/sniffio[${PYTHON_USEDEP}]
	dev-python/redis[${PYTHON_USEDEP}]
	dev-python/mpire[${PYTHON_USEDEP}]
	dev-python/einops[${PYTHON_USEDEP}]
	bsuite-env? ( dev-python/bsuite[${PYTHON_USEDEP}] )
	common-env? ( dev-python/ale-py[${PYTHON_USEDEP}] )
	common-env? ( dev-python/AutoROM[${PYTHON_USEDEP}] )
	common-env? ( dev-python/gym[${PYTHON_USEDEP}] )
	common-env? ( dev-python/cmake[${PYTHON_USEDEP}] )
	common-env? ( dev-python/opencv-python[${PYTHON_USEDEP}] )
	dist? ( dev-python/redis-py-cluster[${PYTHON_USEDEP}] )
	envpool? ( dev-python/envpool[${PYTHON_USEDEP}] )
	fast? ( dev-python/numpy-stl[${PYTHON_USEDEP}] )
	fast? ( dev-python/numba[${PYTHON_USEDEP}] )
	gfootball-env? ( dev-python/gfootball[${PYTHON_USEDEP}] )
	gfootball-env? ( dev-python/kaggle-environments[${PYTHON_USEDEP}] )
	k8s? ( dev-python/kubernetes[${PYTHON_USEDEP}] )
	mario? ( dev-python/gym-super-mario-bros[${PYTHON_USEDEP}] )
	minigrid-env? ( dev-python/minigrid[${PYTHON_USEDEP}] )
	procgen-env? ( dev-python/procgen[${PYTHON_USEDEP}] )
	slimevolleygym-env? ( dev-python/slimevolleygym[${PYTHON_USEDEP}] )
	smac-env? ( dev-python/PySC2[${PYTHON_USEDEP}] )
	sokoban? ( dev-python/gym-sokoban[${PYTHON_USEDEP}] )
	style? ( ~dev-python/yapf-0.29.0[${PYTHON_USEDEP}] )
	style? ( <=dev-python/flake8-3.9.2[${PYTHON_USEDEP}] )"
BDEPEND="${DEPENDENCIES}"
RDEPEND="${DEPENDENCIES}"