Relocate process tree classes

edan-bainglass · Jan 6, 2025 · 33e7517 · 33e7517
1 parent 99be37f
commit 33e7517
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 17 deletions.
diff --git a/src/aiidalab_qe/app/result/components/status/process_tree/__init__.py b/src/aiidalab_qe/app/result/components/status/process_tree/__init__.py
diff --git a/src/aiidalab_qe/app/result/components/status/process_tree/model.py b/src/aiidalab_qe/app/result/components/status/process_tree/model.py
diff --git a/src/aiidalab_qe/app/result/components/status/status.py b/src/aiidalab_qe/app/result/components/status/status.py
@@ -9,7 +9,7 @@
 from aiidalab_widgets_base.viewers import viewer as node_viewer
 
 from .model import WorkChainStatusModel
-from .process_tree import SimplifiedProcessTree, SimplifiedProcessTreeModel
+from .tree import SimplifiedProcessTree, SimplifiedProcessTreeModel
 
 
 class WorkChainStatusPanel(ResultsComponent[WorkChainStatusModel]):

diff --git a/...lt/components/status/process_tree/tree.py → ...b_qe/app/result/components/status/tree.py b/...lt/components/status/process_tree/tree.py → ...b_qe/app/result/components/status/tree.py
@@ -3,12 +3,17 @@
 import typing as t
 
 import ipywidgets as ipw
+import traitlets as tl
 
 from aiida import orm
 from aiida.engine import ProcessState
+from aiidalab_qe.common.mixins import HasProcess
+from aiidalab_qe.common.mvc import Model
 from aiidalab_qe.common.widgets import LoadingWidget
 
-from .model import SimplifiedProcessTreeModel
+
+class SimplifiedProcessTreeModel(Model, HasProcess):
+    clicked = tl.Unicode(None, allow_none=True)
 
 
 class SimplifiedProcessTree(ipw.VBox):