From a3baeab3ea951fea61addb30003236000ce50c07 Mon Sep 17 00:00:00 2001
From: Edan Bainglass <edan.bainglass@gmail.com>
Date: Mon, 6 Jan 2025 05:19:44 +0000
Subject: [PATCH] Relocate process tree classes

---
 .../app/result/components/status/process_tree/__init__.py | 7 -------
 .../app/result/components/status/process_tree/model.py    | 8 --------
 src/aiidalab_qe/app/result/components/status/status.py    | 2 +-
 .../result/components/status/{process_tree => }/tree.py   | 7 ++++++-
 4 files changed, 7 insertions(+), 17 deletions(-)
 delete mode 100644 src/aiidalab_qe/app/result/components/status/process_tree/__init__.py
 delete mode 100644 src/aiidalab_qe/app/result/components/status/process_tree/model.py
 rename src/aiidalab_qe/app/result/components/status/{process_tree => }/tree.py (97%)

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
deleted file mode 100644
index 25be85043..000000000
--- a/src/aiidalab_qe/app/result/components/status/process_tree/__init__.py
+++ /dev/null
@@ -1,7 +0,0 @@
-from .model import SimplifiedProcessTreeModel
-from .tree import SimplifiedProcessTree
-
-__all__ = [
-    "SimplifiedProcessTree",
-    "SimplifiedProcessTreeModel",
-]
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
deleted file mode 100644
index f8db52ff0..000000000
--- a/src/aiidalab_qe/app/result/components/status/process_tree/model.py
+++ /dev/null
@@ -1,8 +0,0 @@
-import traitlets as tl
-
-from aiidalab_qe.common.mixins import HasProcess
-from aiidalab_qe.common.mvc import Model
-
-
-class SimplifiedProcessTreeModel(Model, HasProcess):
-    clicked = tl.Unicode(None, allow_none=True)
diff --git a/src/aiidalab_qe/app/result/components/status/status.py b/src/aiidalab_qe/app/result/components/status/status.py
index 4c1724c06..9ff2892fb 100644
--- 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/src/aiidalab_qe/app/result/components/status/process_tree/tree.py b/src/aiidalab_qe/app/result/components/status/tree.py
similarity index 97%
rename from src/aiidalab_qe/app/result/components/status/process_tree/tree.py
rename to src/aiidalab_qe/app/result/components/status/tree.py
index 6f8d778f3..46cf4b1dc 100644
--- a/src/aiidalab_qe/app/result/components/status/process_tree/tree.py
+++ b/src/aiidalab_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):