add backward capacity

aiidalab · Nov 20, 2023 · 3801b3d · 3801b3d
1 parent a478890
commit 3801b3d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/aiidalab_widgets_base/viewers.py b/aiidalab_widgets_base/viewers.py
@@ -52,8 +52,10 @@ def viewer(obj, **kwargs):
     if key in AIIDA_VIEWER_MAPPING:
         _viewer = AIIDA_VIEWER_MAPPING[key]
         return _viewer(obj, **kwargs)
-    elif isinstance(obj, orm.ProcessNode):
-        return ProcessNodeViewerWidget(obj, **kwargs)
+    # backward capacity
+    elif isinstance(obj, orm.ProcessNode) and obj.node_type in AIIDA_VIEWER_MAPPING:
+        _viewer = AIIDA_VIEWER_MAPPING[key]
+        return _viewer(obj, **kwargs)
     else:
         # No viewer registered for this type, return object itself
         return obj