summaryrefslogtreecommitdiff
path: root/var/spack/repos/builtin/packages/py-tensorboard/package.py
blob: 667ca5d01f2911f0d4378ae523a5359fb16ce827 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
# Copyright 2013-2022 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

from spack.package import *


class PyTensorboard(Package):
    """TensorBoard is a suite of web applications for
    inspecting and understanding your TensorFlow runs and
    graphs."""

    homepage = "https://github.com/tensorflow/tensorboard"
    url      = "https://github.com/tensorflow/tensorboard/archive/2.2.0.tar.gz"

    maintainers = ['aweits']

    version('2.9.0', sha256='c08c5a5d3a88c222ec3419e27bbe41f42ef6d1aac38375b99e9996583ee1f5ae')
    version('2.8.0', sha256='3d6fb62cc38e098679aaace51b55cfaa0478f718668376d47d5e9c616af39329')
    version('2.7.0', sha256='5632812bb9450e5741083b5b7826244ffdb732fb5bce970d526c81874a3e7fb2')
    version('2.6.0', sha256='3d1e0a05828b25c1c28bd90c73d981a0a65c6a5550510bc7983d03ab915e6503')
    version('2.5.0', sha256='58c9e0c31062821ab1c02845c3b7902da92574ef7192d701b1828dacbe4ee610')
    version('2.4.1', sha256='736dc204aa292d221f5871077e60994a9a9ea8e33b841f0d754d510fe6cc7635')
    version('2.4.0', sha256='28a30794c1c797357b2086477394b59afa0b18ca48592ca3c0627f7f10536373')
    version('2.3.0', sha256='947a58702c2841eb4559637dbf8639633f79de9a0f422be9737f3563a1725440')
    version('2.2.0', sha256='d0dfbf0e4b3b5ebbc3fafa6d281d4b9aa5478eac6bac3330652ab6674278ab77')

    depends_on('python@2.7:2.8,3.2:', type=('build', 'run'), when='@:2.5')
    depends_on('python@3.6:', type=('build', 'run'), when='@2.6:')
    # See https://github.com/tensorflow/tensorboard/pull/5528 for upper bounds
    depends_on('bazel@2.1:4', type='build', when='@2.2:2.4')
    depends_on('bazel@3.7:4', type='build', when='@2.5:2.8')
    depends_on('bazel@4:', type='build', when='@2.9:')
    depends_on('py-pip', type='build')
    depends_on('py-absl-py@0.4:', type=('build', 'run'))
    depends_on('py-grpcio@1.24.3:', type=('build', 'run'), when='@2.3:')
    depends_on('py-grpcio@1.23.3:', type=('build', 'run'), when='@2.2')
    depends_on('py-google-auth@1.6.3:1', type=('build', 'run'), when='@:2.6')
    depends_on('py-google-auth@1.6.3:2', type=('build', 'run'), when='@2.7:')
    depends_on('py-google-auth-oauthlib@0.4.1:0.4', type=('build', 'run'))
    depends_on('py-markdown@2.6.8:', 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-protobuf@3.9.2:', type=('build', 'run'), when='@2.9:')
    depends_on('py-requests@2.21.0:2', type=('build', 'run'))
    depends_on('py-setuptools@41.0.0:', type=('build', 'run'))
    depends_on('py-tensorboard-data-server@0.6', type=('build', 'run'), when='@2.5:')
    depends_on('py-tensorboard-plugin-wit@1.6.0:', type=('build', 'run'))
    depends_on('py-werkzeug@0.11.15:', type=('build', 'run'))
    depends_on('py-werkzeug@1.0.1:', type=('build', 'run'), when='@2.9:')
    depends_on('py-wheel@0.26:', type='build')
    depends_on('py-futures@3.1.1:', type=('build', 'run'), when='^python@:2')
    depends_on('py-six@1.10.0:', type=('build', 'run'), when='@:2.4')

    extends('python')

    patch('tboard_shellenv.patch', when='@:2.4')
    patch('webapp.patch', when='@2.7:')
    patch('vz_projector.patch', when='@2.7:')

    phases = ['configure', 'build', 'install']

    # Version 2.6.0 and above do not build in parallel
    @property
    def parallel(self):
        return self.spec.version < Version('2.6.0')

    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):
        self.tmp_path = tempfile.mkdtemp(prefix='spack')
        env.set('TEST_TMPDIR', self.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')
        filter_file('unset PYTHON_HOME',
                    'export PYTHONPATH="{0}"'.format(env['PYTHONPATH']),
                    'tensorboard/pip_package/build_pip_package.sh')
        filter_file('python setup.py',
                    '{0} setup.py'.format(spec['python'].command.path),
                    'tensorboard/pip_package/build_pip_package.sh')
        if self.spec.satisfies('@2.4.0:'):
            filter_file('"@npm//@polymer/polymer",',
                        '"@npm//@polymer/polymer",\n'
                        '"@npm//@types/d3",',
                        'tensorboard/plugins/projector/vz_projector/BUILD')

    def build(self, spec, prefix):
        bazel('--nohome_rc',
              '--nosystem_rc',
              '--output_user_root=' + self.tmp_path,
              'build',
              # watch https://github.com/bazelbuild/bazel/issues/7254
              '--define=EXECUTOR=remote',
              '--verbose_failures',
              '--spawn_strategy=local',
              '--subcommands=pretty_print',
              '--jobs={0}'.format(make_jobs),
              '//tensorboard/pip_package')

    def install(self, spec, prefix):
        with working_dir('spack-build'):
            args = std_pip_args + ['--prefix=' + prefix, '.']
            pip(*args)
        remove_linked_tree(self.tmp_path)