diff --git a/nerfstudio/process_data/equirect_utils.py b/nerfstudio/process_data/equirect_utils.py
index fecf827e4..e7d27c99b 100644
--- a/nerfstudio/process_data/equirect_utils.py
+++ b/nerfstudio/process_data/equirect_utils.py
@@ -252,7 +252,7 @@ def generate_planar_projections_from_equirectangular(
 
     return output_dir, output_mask_dir, output_hdr_dir
 
-def generate_planar_projections_from_equirectangular_2(
+def generate_planar_projections_from_equirectangular_GT(
     metadata_path: Path,
     image_dir: Path,
     planar_image_size: Tuple[int, int],
diff --git a/nerfstudio/scripts/viewer/run_viewer.py b/nerfstudio/scripts/viewer/run_viewer.py
index 56a712026..3e4fe5d3b 100644
--- a/nerfstudio/scripts/viewer/run_viewer.py
+++ b/nerfstudio/scripts/viewer/run_viewer.py
@@ -21,6 +21,7 @@ from __future__ import annotations
 import time
 from dataclasses import dataclass, field, fields
 from pathlib import Path
+
 from typing import Literal
 
 import tyro
@@ -112,9 +113,8 @@ def _start_viewer(config: TrainerConfig, pipeline: Pipeline, step: int):
 
     assert viewer_state and pipeline.datamanager.train_dataset
     viewer_state.init_scene(
-        train_dataset=pipeline.datamanager.train_dataset,
+        dataset=pipeline.datamanager.train_dataset,
         train_state="completed",
-        eval_dataset=pipeline.datamanager.eval_dataset,
     )
     if isinstance(viewer_state, ViewerState):
         viewer_state.viser_server.set_training_state("completed")
@@ -133,4 +133,4 @@ if __name__ == "__main__":
     entrypoint()
 
 # For sphinx docs
-get_parser_fn = lambda: tyro.extras.get_parser(RunViewer)  # noqa
+get_parser_fn = lambda: tyro.extras.get_parser(RunViewer)  # noqa
\ No newline at end of file
diff --git a/nerfstudio/viewer/server/viewer_state.py b/nerfstudio/viewer/server/viewer_state.py
index 03336e8b4..ea0c6c053 100644
--- a/nerfstudio/viewer/server/viewer_state.py
+++ b/nerfstudio/viewer/server/viewer_state.py
@@ -440,4 +440,4 @@ class ViewerState:
     def training_complete(self) -> None:
         """Called when training is complete."""
         self.training_state = "completed"
-        self.viser_server.set_training_state("completed")
+        self.viser_server.set_training_state("completed")
\ No newline at end of file