def wrapper(self, *args, **kwargs):
(logdir, multiplexer) = self.load_runs(run_names)
with self.subTest('bare multiplexer'):
ctx = base_plugin.TBContext(logdir=logdir, multiplexer=multiplexer)
fn(self, histograms_plugin.HistogramsPlugin(ctx), *args, **kwargs)
with self.subTest('generic data provider'):
flags = argparse.Namespace(generic_data='true')
provider = data_provider.MultiplexerDataProvider(multiplexer, logdir)
ctx = base_plugin.TBContext(
flags=flags,
logdir=logdir,
multiplexer=multiplexer,
data_provider=provider,
)
fn(self, histograms_plugin.HistogramsPlugin(ctx), *args, **kwargs)
def __init__(self, context):
"""Instantiates DistributionsPlugin via TensorBoard core.
Args:
context: A base_plugin.TBContext instance.
"""
self._histograms_plugin = histograms_plugin.HistogramsPlugin(context)
def set_up_with_runs(self, run_names):
self.logdir = self.get_temp_dir()
for run_name in run_names:
self.generate_run(run_name)
multiplexer = event_multiplexer.EventMultiplexer(size_guidance={
# don't truncate my test data, please
event_accumulator.TENSORS: self._STEPS,
})
multiplexer.AddRunsFromDirectory(self.logdir)
multiplexer.Reload()
context = base_plugin.TBContext(logdir=self.logdir, multiplexer=multiplexer)
self.plugin = histograms_plugin.HistogramsPlugin(context)
def load_plugin(self, run_names):
(logdir, multiplexer) = self.load_runs(run_names)
provider = data_provider.MultiplexerDataProvider(multiplexer, logdir)
ctx = base_plugin.TBContext(logdir=logdir, data_provider=provider)
return histograms_plugin.HistogramsPlugin(ctx)
