def bootstrap_plugin(self):
multiplexer = event_multiplexer.EventMultiplexer()
multiplexer.AddRunsFromDirectory(self.logdir)
multiplexer.Reload()
context = base_plugin.TBContext(logdir=self.logdir,
multiplexer=multiplexer)
self.plugin = graphs_plugin.GraphsPlugin(context)
def wrapper(self, *args, **kwargs):
(logdir, multiplexer) = self.load_runs(run_specs)
with self.subTest("bare multiplexer"):
ctx = base_plugin.TBContext(logdir=logdir,
multiplexer=multiplexer)
fn(self, graphs_plugin.GraphsPlugin(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, graphs_plugin.GraphsPlugin(ctx), *args, **kwargs)
def set_up_with_runs(self, with_graph=True, without_graph=True):
self.logdir = self.get_temp_dir()
if with_graph:
self.generate_run(self._RUN_WITH_GRAPH, include_graph=True)
if without_graph:
self.generate_run(self._RUN_WITHOUT_GRAPH, include_graph=False)
multiplexer = event_multiplexer.EventMultiplexer()
multiplexer.AddRunsFromDirectory(self.logdir)
multiplexer.Reload()
context = base_plugin.TBContext(logdir=self.logdir, multiplexer=multiplexer)
self.plugin = graphs_plugin.GraphsPlugin(context)
def load_plugin(self, run_specs):
logdir = self.get_temp_dir()
for run_spec in run_specs:
self.generate_run(logdir, *run_spec)
multiplexer = event_multiplexer.EventMultiplexer()
multiplexer.AddRunsFromDirectory(logdir)
multiplexer.Reload()
provider = data_provider.MultiplexerDataProvider(multiplexer, logdir)
ctx = base_plugin.TBContext(
logdir=logdir,
multiplexer=multiplexer,
data_provider=provider,
)
return graphs_plugin.GraphsPlugin(ctx)
