# 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="Petastorm is a library enabling the use of Parquet storage from Tensorflow, Pytorch, and other Python-based ML training frameworks. [top-max 0.13.1]"

HOMEPAGE="https://github.com/uber/petastorm"
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="docs opencv s3fs test tf tf_gpu torch"
DEPENDENCIES=">=dev-python/dill-0.2.1[${PYTHON_USEDEP}]
	>=dev-python/diskcache-3.0.0[${PYTHON_USEDEP}]
	dev-python/future[${PYTHON_USEDEP}]
	>=dev-python/numpy-1.13.3[${PYTHON_USEDEP}]
	>=dev-python/packaging-15.0[${PYTHON_USEDEP}]
	>=dev-python/pandas-0.19.0[${PYTHON_USEDEP}]
	>=dev-python/psutil-4.0.0[${PYTHON_USEDEP}]
	dev-python/pyspark[${PYTHON_USEDEP}]
	>=dev-python/pyzmq-14.0.0[${PYTHON_USEDEP}]
	>=dev-python/pyarrow-0.17.1[${PYTHON_USEDEP}]
	>=dev-python/six-1.5.0[${PYTHON_USEDEP}]
	dev-python/fsspec[${PYTHON_USEDEP}]
	docs? ( >=dev-python/sphinx-1.2.2[${PYTHON_USEDEP}] )
	docs? ( >=dev-python/alabaster-0.7.11[${PYTHON_USEDEP}] )
	opencv? ( dev-python/opencv-python[${PYTHON_USEDEP}] )
	s3fs? ( dev-python/s3fs[${PYTHON_USEDEP}] )
	test? ( >=dev-python/pillow-6.2.1[${PYTHON_USEDEP}] )
	test? ( dev-python/codecov[${PYTHON_USEDEP}] )
	test? ( dev-python/flake8[${PYTHON_USEDEP}] )
	test? ( dev-python/gcsfs[${PYTHON_USEDEP}] )
	test? ( >=dev-python/mock-2.0.0[${PYTHON_USEDEP}] )
	test? ( dev-python/mypy[${PYTHON_USEDEP}] )
	test? ( dev-python/opencv-python[${PYTHON_USEDEP}] )
	test? ( >=dev-python/pylint-1.9[${PYTHON_USEDEP}] )
	test? ( >=dev-python/pytest-3.0.0[${PYTHON_USEDEP}] )
	test? ( >=dev-python/pytest-cov-2.5.1[${PYTHON_USEDEP}] )
	test? ( >=dev-python/pytest-forked-0.2[${PYTHON_USEDEP}] )
	test? ( dev-python/pytest-logger[${PYTHON_USEDEP}] )
	test? ( >=dev-python/pytest-timeout-1.3.3[${PYTHON_USEDEP}] )
	test? ( >=dev-python/requests-2.22.0[${PYTHON_USEDEP}] )
	test? ( dev-python/s3fs[${PYTHON_USEDEP}] )
	test? ( dev-python/gcsfs[${PYTHON_USEDEP}] )
	tf? ( dev-python/tensorflow[${PYTHON_USEDEP}] )
	tf_gpu? ( dev-python/tensorflow-gpu[${PYTHON_USEDEP}] )
	torch? ( dev-python/torchvision[${PYTHON_USEDEP}] )
	torch? ( dev-python/torch[${PYTHON_USEDEP}] )"
BDEPEND="${DEPENDENCIES}"
RDEPEND="${DEPENDENCIES}"