diff options
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/py-tensorflow-hub/package.py | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-tensorflow-hub/package.py b/var/spack/repos/builtin/packages/py-tensorflow-hub/package.py new file mode 100644 index 0000000000..f432c8fc8c --- /dev/null +++ b/var/spack/repos/builtin/packages/py-tensorflow-hub/package.py @@ -0,0 +1,75 @@ +# Copyright 2013-2021 Lawrence Livermore National Security, LLC and other +# Spack Project Developers. See the top-level COPYRIGHT file for details. +# +# SPDX-License-Identifier: (Apache-2.0 OR MIT) + +import tempfile + + +class PyTensorflowHub(Package): + """TensorFlow Hub is a library to foster the publication, discovery, and + consumption of reusable parts of machine learning models.""" + + homepage = "https://github.com/tensorflow/hub" + url = "https://github.com/tensorflow/hub/archive/refs/tags/v0.12.0.tar.gz" + + maintainers = ['aweits'] + + version('0.12.0', sha256='b192ef3a9a6cbeaee46142d64b47b979828dbf41fc56d48c6587e08f6b596446') + version('0.11.0', sha256='4715a4212b45531a7c25ada7207d850467d1b5480f1940f16623f8770ad64df4') + + extends('python') + + depends_on('bazel', type='build') + depends_on('py-setuptools', type='build') + depends_on('python@3.6:', type=('build', 'run')) + depends_on('py-numpy@1.12.0:', type=('build', 'run')) + depends_on('py-protobuf@3.8.0:', type=('build', 'run')) + + def install(self, spec, prefix): + tmp_path = tempfile.mkdtemp(prefix='spack') + env['TEST_TMPDIR'] = tmp_path + env['HOME'] = tmp_path + args = [ + # Don't allow user or system .bazelrc to override build settings + '--nohome_rc', + '--nosystem_rc', + # Bazel does not work properly on NFS, switch to /tmp + '--output_user_root=' + tmp_path, + 'build', + # Spack logs don't handle colored output well + '--color=no', + '--jobs={0}'.format(make_jobs), + # Enable verbose output for failures + '--verbose_failures', + # Show (formatted) subcommands being executed + '--subcommands=pretty_print', + '--spawn_strategy=local', + # Ask bazel to explain what it's up to + # Needs a filename as argument + '--explain=explainlogfile.txt', + # Increase verbosity of explanation, + '--verbose_explanations', + # bazel uses system PYTHONPATH instead of spack paths + '--action_env', 'PYTHONPATH={0}'.format(env['PYTHONPATH']), + '//tensorflow_hub/pip_package:build_pip_package', + ] + + bazel(*args) + + runfiles = join_path('bazel-bin', 'tensorflow_hub', 'pip_package', + 'build_pip_package.runfiles', 'org_tensorflow_hub') + insttmp_path = tempfile.mkdtemp(prefix='spack') + install(join_path('tensorflow_hub', 'pip_package', 'setup.py'), insttmp_path) + install(join_path('tensorflow_hub', 'pip_package', 'setup.cfg'), insttmp_path) + install('LICENSE', join_path(insttmp_path, 'LICENSE.txt')) + mkdirp(join_path(insttmp_path, 'tensorflow_hub')) + install_tree(join_path(runfiles, 'tensorflow_hub'), + join_path(insttmp_path, 'tensorflow_hub')) + + with working_dir(insttmp_path): + setup_py('install', '--prefix={0}'.format(prefix), + '--single-version-externally-managed', '--root=/') + + remove_linked_tree(tmp_path) + remove_linked_tree(insttmp_path) |