From 43703fa05a306d765b5c6c57ccc763af3de66f0b Mon Sep 17 00:00:00 2001 From: Chung Min Kim Date: Sun, 26 Nov 2023 21:56:49 -0800 Subject: [PATCH] Bugfix; scrape pipeline for viewercontrol, not trainer (#2621) Scrape pipeline for viewercontrol, not trainer --- nerfstudio/viewer_beta/viewer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/viewer_beta/viewer.py b/nerfstudio/viewer_beta/viewer.py index 3054cf5cb..e577ca5c3 100644 --- a/nerfstudio/viewer_beta/viewer.py +++ b/nerfstudio/viewer_beta/viewer.py @@ -213,7 +213,7 @@ class Viewer: # scrape the trainer/pipeline for any ViewerControl objects to initialize them self.viewer_controls: List[ViewerControl] = [ - e for (_, e) in parse_object(self.trainer, ViewerControl, "Custom Elements") + e for (_, e) in parse_object(pipeline, ViewerControl, "Custom Elements") ] for c in self.viewer_controls: c._setup(self)