# 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="A 2D and 3D PyTorch implementation of the Tiramisu CNN"

HOMEPAGE="https://github.com/jcreinhold/tiramisu-brulee"
LICENSE="Apache Software License 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="lesionseg mlflow onnx"
DEPENDENCIES="dev-python/torch[${PYTHON_USEDEP}]
	lesionseg? ( dev-python/jsonargparse[${PYTHON_USEDEP}] )
	lesionseg? ( dev-python/numpy[${PYTHON_USEDEP}] )
	lesionseg? ( dev-python/pandas[${PYTHON_USEDEP}] )
	lesionseg? ( >=dev-python/pillow-9.0.0[${PYTHON_USEDEP}] )
	lesionseg? ( dev-python/pytorch-lightning[${PYTHON_USEDEP}] )
	lesionseg? ( dev-python/pyyaml[${PYTHON_USEDEP}] )
	lesionseg? ( dev-python/ruyaml[${PYTHON_USEDEP}] )
	lesionseg? ( dev-python/scikit-image[${PYTHON_USEDEP}] )
	lesionseg? ( dev-python/scipy[${PYTHON_USEDEP}] )
	lesionseg? ( dev-python/SimpleITK[${PYTHON_USEDEP}] )
	lesionseg? ( dev-python/torchio[${PYTHON_USEDEP}] )
	lesionseg? ( dev-python/torchmetrics[${PYTHON_USEDEP}] )
	mlflow? ( dev-python/mlflow[${PYTHON_USEDEP}] )
	onnx? ( dev-python/onnx[${PYTHON_USEDEP}] )
	onnx? ( dev-python/onnxruntime[${PYTHON_USEDEP}] )"
BDEPEND="${DEPENDENCIES}"
RDEPEND="${DEPENDENCIES}"