nit

Signed-off-by: Future-Outlier <[email protected]>
flyteorg · Apr 6, 2024 · 028667b · 028667b
1 parent eed70f2
commit 028667b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/flytekit/core/workflow.py b/flytekit/core/workflow.py
@@ -288,9 +288,8 @@ def __call__(self, *args, **kwargs) -> Union[Tuple[Promise], Promise, VoidPromis
         except Exception as exc:
             if self.on_failure:
                 if self.on_failure.python_interface and "err" in self.on_failure.python_interface.inputs:
-                    input_kwargs["err"] = FlyteError(
-                        failed_node_id=self.failure_node.id if self.failure_node else "", message=str(exc)
-                    )
+                    id = self.failure_node.id if self.failure_node else ""
+                    input_kwargs["err"] = FlyteError(failed_node_id=id, message=str(exc))
                 self.on_failure(**input_kwargs)
             raise exc