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)