# 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="Pyro PPL on NumPy"

HOMEPAGE="https://github.com/pyro-ppl/numpyro"
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="cpu cuda dev doc examples test tpu"
DEPENDENCIES="dev-python/jax[${PYTHON_USEDEP}]
	dev-python/jaxlib[${PYTHON_USEDEP}]
	dev-python/multipledispatch[${PYTHON_USEDEP}]
	dev-python/numpy[${PYTHON_USEDEP}]
	dev-python/tqdm[${PYTHON_USEDEP}]
	doc? ( dev-python/ipython[${PYTHON_USEDEP}] )
	doc? ( >=dev-python/nbsphinx-0.8.9[${PYTHON_USEDEP}] )
	doc? ( dev-python/readthedocs-sphinx-search[${PYTHON_USEDEP}] )
	doc? ( >=dev-python/sphinx-5.0[${PYTHON_USEDEP}] )
	doc? ( dev-python/sphinx-rtd-theme[${PYTHON_USEDEP}] )
	doc? ( dev-python/sphinx-gallery[${PYTHON_USEDEP}] )
	test? ( <dev-python/importlib-metadata-5.0[${PYTHON_USEDEP}] )
	test? ( dev-python/ruff[${PYTHON_USEDEP}] )
	test? ( >=dev-python/mypy-1.13[${PYTHON_USEDEP}] )
	test? ( >=dev-python/pytest-4.1[${PYTHON_USEDEP}] )
	test? ( dev-python/pyro-api[${PYTHON_USEDEP}] )
	test? ( dev-python/scikit-learn[${PYTHON_USEDEP}] )
	test? ( >=dev-python/scipy-1.9[${PYTHON_USEDEP}] )
	dev? ( dev-python/dm-haiku[${PYTHON_USEDEP}] )
	dev? ( dev-python/equinox[${PYTHON_USEDEP}] )
	dev? ( dev-python/flax[${PYTHON_USEDEP}] )
	dev? ( dev-python/funsor[${PYTHON_USEDEP}] )
	dev? ( dev-python/graphviz[${PYTHON_USEDEP}] )
	dev? ( dev-python/jaxns[${PYTHON_USEDEP}] )
	dev? ( dev-python/matplotlib[${PYTHON_USEDEP}] )
	dev? ( dev-python/optax[${PYTHON_USEDEP}] )
	dev? ( dev-python/pylab-sdk[${PYTHON_USEDEP}] )
	dev? ( dev-python/pytest-cov[${PYTHON_USEDEP}] )
	dev? ( dev-python/pyyaml[${PYTHON_USEDEP}] )
	dev? ( dev-python/requests[${PYTHON_USEDEP}] )
	dev? ( dev-python/tfp-nightly[${PYTHON_USEDEP}] )
	examples? ( dev-python/arviz[${PYTHON_USEDEP}] )
	examples? ( dev-python/jupyter[${PYTHON_USEDEP}] )
	examples? ( dev-python/matplotlib[${PYTHON_USEDEP}] )
	examples? ( dev-python/pandas[${PYTHON_USEDEP}] )
	examples? ( dev-python/seaborn[${PYTHON_USEDEP}] )
	examples? ( dev-python/scikit-learn[${PYTHON_USEDEP}] )
	examples? ( dev-python/wordcloud[${PYTHON_USEDEP}] )
	cpu? ( dev-python/jax[${PYTHON_USEDEP}] )
	tpu? ( dev-python/jax[${PYTHON_USEDEP}] )
	cuda? ( dev-python/jax[${PYTHON_USEDEP}] )"
BDEPEND="${DEPENDENCIES}"
RDEPEND="${DEPENDENCIES}"