diff --git a/taipy/gui_core/_adapters.py b/taipy/gui_core/_adapters.py index 7cb5defe1d..b48a2ceb4f 100644 --- 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,