From f32f9778c986bc0fd6c1dca316cfc96fd46d26cf Mon Sep 17 00:00:00 2001 From: Jan Janssen Date: Wed, 2 Apr 2025 12:28:22 +0200 Subject: [PATCH] Bug fix for pyiron_base interface --- .../src/python_workflow_definition/pyiron_base.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python_workflow_definition/src/python_workflow_definition/pyiron_base.py b/python_workflow_definition/src/python_workflow_definition/pyiron_base.py index 016514a..0e31924 100644 --- a/python_workflow_definition/src/python_workflow_definition/pyiron_base.py +++ b/python_workflow_definition/src/python_workflow_definition/pyiron_base.py @@ -39,12 +39,13 @@ def _group_edges(edges_lst): def _get_source(nodes_dict, delayed_object_dict, source, sourceHandle): - if source in delayed_object_dict.keys(): + if source in delayed_object_dict.keys() and sourceHandle is not None: return delayed_object_dict[source].__getattr__("output").__getattr__(sourceHandle) + elif source in delayed_object_dict.keys(): + return delayed_object_dict[source] else: return nodes_dict[source] - def _get_delayed_object_dict(total_lst, nodes_dict, source_handle_dict, pyiron_project): delayed_object_dict = {} for item in total_lst: