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)