spack/var/spack/repos/builtin/packages/py-tensorboard/package.py
Andrew W Elble 0df70e217f
A collection of tensorflow fixes and updates (#19371)
* A collection of tensorflow fixes and updates

* tensorflow 2.3.1 requires the workaround for external protobuf as well
* Update tensorflow-estimator to 2.3.0
* Update tensorboard to 2.3.0
* Update tensorboard-plugin-wit to use actual releases
* Patch that potentially fixes #16073

* add myself to maintainer list
2020-10-19 13:15:49 -05:00

94 lines
4.0 KiB
Python

# Copyright 2013-2020 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)
class PyTensorboard(Package):
"""TensorBoard is a suite of web applications for
inspecting and understanding your TensorFlow runs and
graphs."""
homepage = "https://pypi.python.org/project/tensorboard"
url = "https://github.com/tensorflow/tensorboard/archive/2.2.0.tar.gz"
maintainers = ['aweits']
version('2.3.0', sha256='947a58702c2841eb4559637dbf8639633f79de9a0f422be9737f3563a1725440')
version('2.2.0', sha256='d0dfbf0e4b3b5ebbc3fafa6d281d4b9aa5478eac6bac3330652ab6674278ab77')
depends_on('python@2.7:2.8,3.2:', type=('build', 'run'))
depends_on('bazel@0.26.1:', type='build')
depends_on('py-setuptools@41.0.0:', type=('build', 'run'))
depends_on('py-absl-py@0.4:', type=('build', 'run'))
depends_on('py-markdown@2.6.8:', type=('build', 'run'))
depends_on('py-requests@2.21.0:2.999', type=('build', 'run'))
depends_on('py-futures@3.1.1:', type=('build', 'run'), when='^python@:2')
depends_on('py-grpcio@1.24.3:', type=('build', 'run'), when='@2.3.0')
depends_on('py-grpcio@1.23.3:', type=('build', 'run'), when='@2.2.0')
depends_on('py-google-auth@1.6.3:1.99.99', type=('build', 'run'))
depends_on('py-numpy@1.12.0:', type=('build', 'run'))
depends_on('py-protobuf@3.6.0:', type=('build', 'run'))
depends_on('py-six@1.10.0:', type=('build', 'run'))
depends_on('py-werkzeug@0.11.15:', type=('build', 'run'))
depends_on('py-wheel', type='build')
depends_on('py-wheel@0.26:', type='build', when='@0.6: ^python@3:')
depends_on('py-google-auth-oauthlib@0.4.1:0.4.999', type=('build', 'run'))
depends_on('py-tensorboard-plugin-wit@1.6.0:', type=('build', 'run'), when='@2.2.0:')
depends_on('py-tensorflow-estimator@2.2.0', type='run', when='@2.2.0')
depends_on('py-tensorflow-estimator@2.3.0', type='run', when='@2.3.0')
extends('python')
patch('tboard_shellenv.patch')
phases = ['configure', 'build', 'install']
def patch(self):
filter_file('build --define=angular_ivy_enabled=True',
'build --define=angular_ivy_enabled=True\n'
'build --distinct_host_configuration=false\n'
'build --action_env=PYTHONPATH="{0}"\n'.format(
env['PYTHONPATH']),
'.bazelrc')
def setup_build_environment(self, env):
tmp_path = '/tmp/spack/tb'
mkdirp(tmp_path)
env.set('TEST_TMPDIR', tmp_path)
def configure(self, spec, prefix):
builddir = join_path(self.stage.source_path, 'spack-build')
mkdirp(builddir)
filter_file(r'workdir=.*',
'workdir="{0}"'.format(builddir),
'tensorboard/pip_package/build_pip_package.sh')
filter_file(r'pip install .*',
'',
'tensorboard/pip_package/build_pip_package.sh')
filter_file(r'command \-v .*',
'',
'tensorboard/pip_package/build_pip_package.sh')
filter_file(r'virtualenv .*',
'',
'tensorboard/pip_package/build_pip_package.sh')
filter_file('trap cleanup EXIT',
'',
'tensorboard/pip_package/build_pip_package.sh')
def build(self, spec, prefix):
tmp_path = env['TEST_TMPDIR']
bazel('--nohome_rc',
'--nosystem_rc',
'--output_user_root=' + tmp_path,
'build',
'--verbose_failures',
'--spawn_strategy=local',
'--subcommands=pretty_print',
'//tensorboard/pip_package')
def install(self, spec, prefix):
with working_dir('spack-build'):
setup_py('install', '--prefix={0}'.format(prefix),
'--single-version-externally-managed', '--root=/')