From f5fa186ae57d0418ec9f480df46cd164d6787128 Mon Sep 17 00:00:00 2001 From: superstar54 Date: Tue, 28 Jan 2025 16:26:52 +0100 Subject: [PATCH] delete workgraph related attributes of the func if exist --- src/aiida_workgraph/engine/utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/aiida_workgraph/engine/utils.py b/src/aiida_workgraph/engine/utils.py index f4dfeb3c..8de91b1f 100644 --- a/src/aiida_workgraph/engine/utils.py +++ b/src/aiida_workgraph/engine/utils.py @@ -84,7 +84,10 @@ def prepare_for_python_task(task: dict, kwargs: dict, var_kwargs: dict) -> dict: function_outputs.append( {"name": output["name"], "identifier": output["identifier"]} ) - + # delete workgraph related attributes of the func if exist + for attr in ["task", "tdata", "node"]: + if hasattr(func, attr): + delattr(func, attr) inputs = prepare_pythonjob_inputs( function=func, function_inputs=function_inputs,