Fix element parsing (#1803)

* fix trainer element hierarchy
This commit is contained in:
Justin Kerr authored and GitHub committed 2023-04-23 17:42:37 -07:00
1 parent 09f56ab00d
commit c256dc2bf9
1 file changed
+2 -2
+2 -2
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)