diff options
Diffstat (limited to 'sci-libs/tensorflow-estimator/tensorflow-estimator-2.12.0.ebuild')
-rw-r--r-- | sci-libs/tensorflow-estimator/tensorflow-estimator-2.12.0.ebuild | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/sci-libs/tensorflow-estimator/tensorflow-estimator-2.12.0.ebuild b/sci-libs/tensorflow-estimator/tensorflow-estimator-2.12.0.ebuild new file mode 100644 index 000000000000..fbe71a0d54d6 --- /dev/null +++ b/sci-libs/tensorflow-estimator/tensorflow-estimator-2.12.0.ebuild @@ -0,0 +1,75 @@ +# Copyright 1999-2023 Gentoo Authors +# Distributed under the terms of the GNU General Public License v2 + +EAPI=8 + +PYTHON_COMPAT=( python3_10 python3_11 ) +MY_PN="estimator" +MY_PV=${PV/_rc/-rc} +MY_P=${MY_PN}-${MY_PV} + +inherit bazel distutils-r1 + +DESCRIPTION="A high-level TensorFlow API that greatly simplifies machine learning programming" +HOMEPAGE="https://www.tensorflow.org/" + +LICENSE="Apache-2.0" +SLOT="0" +KEYWORDS="~amd64" +IUSE="" + +bazel_external_uris=" + https://github.com/bazelbuild/rules_cc/archive/b1c40e1de81913a3c40e5948f78719c28152486d.zip -> bazelbuild-rules_cc-b1c40e1de81913a3c40e5948f78719c28152486d.zip + https://github.com/bazelbuild/rules_java/archive/7cf3cefd652008d0a64a419c34c13bdca6c8f178.zip -> bazelbuild-rules_java-7cf3cefd652008d0a64a419c34c13bdca6c8f178.zip" + +SRC_URI="https://github.com/tensorflow/${MY_PN}/archive/v${MY_PV}.tar.gz -> ${P}.tar.gz + ${bazel_external_uris}" + +RDEPEND=" + sci-libs/tensorflow[python,${PYTHON_USEDEP}] + sci-libs/keras[${PYTHON_USEDEP}]" +DEPEND="${RDEPEND}" +BDEPEND=" + app-arch/unzip + dev-java/java-config + >=dev-util/bazel-5.3.0" + +S="${WORKDIR}/${MY_P}" + +DOCS=( CONTRIBUTING.md README.md ) + +src_unpack() { + unpack "${P}.tar.gz" + bazel_load_distfiles "${bazel_external_uris}" +} + +src_prepare() { + bazel_setup_bazelrc + default + python_copy_sources +} + +python_compile() { + pushd "${BUILD_DIR}" >/dev/null || die + + ebazel build //tensorflow_estimator/tools/pip_package:build_pip_package + ebazel shutdown + + local srcdir="${T}/src-${EPYTHON/./_}" + mkdir -p "${srcdir}" || die + bazel-bin/tensorflow_estimator/tools/pip_package/build_pip_package --src "${srcdir}" || die + + popd >/dev/null || die +} + +src_compile() { + export JAVA_HOME=$(java-config --jre-home) + distutils-r1_src_compile +} + +python_install() { + pushd "${T}/src-${EPYTHON/./_}" >/dev/null || die + esetup.py install + python_optimize + popd >/dev/null || die +} |