def get_run_visualization_dict(self, trial_type):
try:
return self.profile_dict["run_visualization_dict"][trial_type]
except:
logging.info("You have never visualized runs of this trial type, will update your profile using default configuration")
self.update_run_visualization_dict(trial_type, get_run_type_visualizer(trial_type)["default"].get_default_attributes())
return self.profile_dict["run_visualization_dict"][trial_type]
def update(self, trial=None, run=None, visualize=None):
if visualize:
if not (trial is None):
self.plot_view = get_trial_type_visualizer(trial.get_trial_type())["default"]
if trial.get_main_counter() % trial.get_configuration().vis_every_X_steps == 0: ## TODO - its not ideal... rethink it...
snapshot = trial.snapshot()
graphdict = self.controller.get_trial_visualization_dict(trial.get_trial_type())
snapshot.update(graphdict)
self.controller.visualize(snapshot, self.plot_view.render)
if not (run is None):
self.run_view = get_run_type_visualizer(run.get_trial_type())["default"]
logging.info("Generating Report")
snapshot = run.snapshot()
snapshot.update(self.controller.get_run_visualization_dict(run.get_run_type()))
self.controller.visualize(snapshot, self.run_view.render)
