Bugfix; scrape pipeline for viewercontrol, not trainer (#2621)

Scrape pipeline for viewercontrol, not trainer
This commit is contained in:
Chung Min Kim authored and GitHub committed 2023-11-26 21:56:49 -08:00
1 parent 9528a3f717
commit 43703fa05a
1 file changed
+1 -1
+1 -1
View File
@@ -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)