Show additional data nodes in DAG (#1771)

* Show additional data nodes in DAG resolves #787 * E501 * E501 --------- Co-authored-by: Fred Lefévère-Laoide <[email protected]>
Avaiga · Sep 10, 2024 · aae5778 · aae5778
1 parent 042d04e
commit aae5778
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/taipy/gui_core/_adapters.py b/taipy/gui_core/_adapters.py
@@ -131,16 +131,28 @@ def get(self):
                 if scenario := core_get(data.id):
                     dag = scenario._get_dag()
                     nodes = {}
-                    for id, node in dag.nodes.items():
-                        entityType = _GuiCoreScenarioDagAdapter.get_entity_type(node)
+                    for id, dag_node in dag.nodes.items():
+                        entityType = _GuiCoreScenarioDagAdapter.get_entity_type(dag_node)
                         cat = nodes.get(entityType)
                         if cat is None:
                             cat = {}
                             nodes[entityType] = cat
                         cat[id] = {
-                            "name": node.entity.get_simple_label(),
-                            "type": node.entity.storage_type() if hasattr(node.entity, "storage_type") else None,
+                            "name": dag_node.entity.get_simple_label(),
+                            "type": dag_node.entity.storage_type()
+                            if hasattr(dag_node.entity, "storage_type")
+                            else None,
                         }
+                    cat = nodes.get(DataNode.__name__)
+                    if cat is None:
+                        cat = {}
+                        nodes[DataNode.__name__] = cat
+                    for id, data_node in scenario.additional_data_nodes.items():
+                        cat[id] = {
+                            "name": data_node.get_simple_label(),
+                            "type": data_node.storage_type(),
+                        }
+
                     return [
                         data.id,
                         nodes,