From cc494adb2724d8a8ffe4f51a9d1bc17a915a2ca1 Mon Sep 17 00:00:00 2001 From: SeungWoo Yoo Date: Thu, 20 Apr 2023 13:47:34 +0900 Subject: [PATCH] Set data path to launch viewer for DreamFusion (#1776) * Set data path to launch viewer for DreamFusion * Apply formatter * Fallback to 'base_dir' if data manager has no path * Make if statement non-inline * Reformat --------- Co-authored-by: Matthew Tancik --- scripts/viewer/run_viewer.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/scripts/viewer/run_viewer.py b/scripts/viewer/run_viewer.py index c48b23d9b..6c092f1cd 100644 --- a/scripts/viewer/run_viewer.py +++ b/scripts/viewer/run_viewer.py @@ -61,8 +61,15 @@ class RunViewer: def _start_viewer(self, config: TrainerConfig, pipeline: Pipeline): base_dir = config.get_base_dir() viewer_log_path = base_dir / config.viewer.relative_log_filename + + datapath = pipeline.datamanager.get_datapath() + if datapath is None: + datapath = base_dir + viewer_state, banner_messages = viewer_utils.setup_viewer( - config.viewer, log_filename=viewer_log_path, datapath=pipeline.datamanager.get_datapath() + config.viewer, + log_filename=viewer_log_path, + datapath=datapath, ) # We don't need logging, but writer.GLOBAL_BUFFER needs to be populated