delete workgraph related attributes of the func if exist

aiidateam · Jan 28, 2025 · f5fa186 · f5fa186
1 parent 34e32bd
commit f5fa186
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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,