Default nb task resolver msg (#2889)

* add additional info to default notebook task resolver Signed-off-by: Niels Bantilan <[email protected]> * add object repr to error message Signed-off-by: Niels Bantilan <[email protected]> * update string Signed-off-by: Niels Bantilan <[email protected]> --------- Signed-off-by: Niels Bantilan <[email protected]>
flyteorg · Nov 1, 2024 · 804dae1 · 804dae1
1 parent 5bf0acd
commit 804dae1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/flytekit/core/python_auto_container.py b/flytekit/core/python_auto_container.py
@@ -339,7 +339,8 @@ def load_task(self, loader_args: List[str]) -> PythonAutoContainerTask:
         # verify the loaded_data is of the correct type
         if not isinstance(loaded_data, PickledEntity):
             raise RuntimeError(
-                "The loaded data is not of the correct type. Please ensure that the pickle file is not corrupted."
+                f"The loaded data is not of the correct type. Expected PickledEntity, found {type(loaded_data)}. "
+                f"Please ensure that the pickle file is not corrupted. Loaded data: {loaded_data}"
             )
         pickled_object: PickledEntity = loaded_data