diff options
Diffstat (limited to 'sci-libs/tensorflow-estimator/tensorflow-estimator-1.13.0.ebuild')
-rw-r--r-- | sci-libs/tensorflow-estimator/tensorflow-estimator-1.13.0.ebuild | 61 |
1 files changed, 61 insertions, 0 deletions
diff --git a/sci-libs/tensorflow-estimator/tensorflow-estimator-1.13.0.ebuild b/sci-libs/tensorflow-estimator/tensorflow-estimator-1.13.0.ebuild new file mode 100644 index 000000000000..a5aede9bca28 --- /dev/null +++ b/sci-libs/tensorflow-estimator/tensorflow-estimator-1.13.0.ebuild @@ -0,0 +1,61 @@ +# Copyright 1999-2019 Jason Zaman +# Distributed under the terms of the GNU General Public License v2 + +EAPI=7 + +PYTHON_COMPAT=( python2_7 python{3_5,3_6,3_7} ) +MY_PN="estimator" +MY_PV=${PV/_rc/-rc} +MY_P=${MY_PN}-${MY_PV} + +inherit bazel distutils-r1 flag-o-matic toolchain-funcs + +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="" + +SRC_URI="https://github.com/tensorflow/${MY_PN}/archive/v${MY_PV}.tar.gz -> ${P}.tar.gz" + +RDEPEND="sci-libs/tensorflow[python,${PYTHON_USEDEP}]" +DEPEND="${RDEPEND}" +BDEPEND=" + dev-java/java-config" + +S="${WORKDIR}/${MY_P}" + +PATCHES=( + "${FILESDIR}/0001-pip_package-modularize-build-script-to-allow-distros.patch" +) +DOCS=( CONTRIBUTING.md README.md ) + +src_prepare() { + bazel_setup_bazelrc + default +} + +src_compile() { + export JAVA_HOME=$(java-config --jre-home) + + ebazel build //tensorflow_estimator/tools/pip_package:build_pip_package + ebazel shutdown + + local srcdir="${T}/src" + mkdir -p "${srcdir}" || die + bazel-bin/tensorflow_estimator/tools/pip_package/build_pip_package --src "${srcdir}" || die +} + +src_install() { + do_install() { + cd "${T}/src" || die + esetup.py install + python_optimize + } + python_foreach_impl do_install + + cd "${S}" || die + einstalldocs +} |