4
0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00

Fix element parsing (#1803)

* fix trainer element hierarchy
This commit is contained in:
Justin Kerr 2023-04-23 17:42:37 -07:00 committed by GitHub
parent 09f56ab00d
commit c256dc2bf9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -152,8 +152,8 @@ class ViewerState:
self.viewer_elements = []
if self.trainer is not None:
self.viewer_elements.extend(parse_object(self.trainer, ViewerElement, "Trainer"))
self.viewer_elements.extend(parse_object(pipeline, ViewerElement, "Pipeline"))
else:
self.viewer_elements.extend(parse_object(pipeline, ViewerElement, "Pipeline"))
for param_path, element in self.viewer_elements:
folder_labels = param_path.split("/")[:-1]
nested_folder_install(folder_labels, element)