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