fix failure node example code (#1778)

Signed-off-by: Alex Wu <[email protected]>

examples/data_types_and_io/data_types_and_io/pickle_type.py

@@ -34,7 +34,7 @@ def superhero_wf(name: str = "Thor", power: str = "Flight") -> str:
 # The following example demonstrates how to set the batch size.
 from typing import Iterator
 
-from flytekit.types.pickle.pickle import BatchSize
+from flytekit.core.type_engine import BatchSize
 from typing_extensions import Annotated
 
 

examples/development_lifecycle/development_lifecycle/failure_node.py

@@ -1,4 +1,7 @@
+import typing
+
 from flytekit import WorkflowFailurePolicy, task, workflow
+from flytekit.types.error.error import FlyteError
 
 
 @task
@@ -20,8 +23,8 @@ def delete_cluster(name: str):
 
 # Create a task that will be executed if any of the tasks in the workflow fail
 @task
-def clean_up(name: str):
-    print(f"Cleaning up cluster {name}")
+def clean_up(name: str, err: typing.Optional[FlyteError] = None):
+    print(f"Deleting cluster {name} due to {err}")
 
 
 # Specify the `on_failure` to a cleanup task.