Skip to content

Commit d57afae

Browse files
committed
better cast
Signed-off-by: dafnapension <dafnashein@yahoo.com>
1 parent 041bf52 commit d57afae

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/unitxt/error_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -272,7 +272,7 @@ def _add_context_to_exception(
272272
context_parts = _build_context_parts(final_context_object, final_context)
273273
context_message = _create_context_box(context_parts)
274274
if type(original_error) == KeyError:
275-
f = ExtKeyError(original_message)
275+
f = ExtKeyError(KeyError(original_error))
276276
else:
277277
f = original_error
278278
_store_context_attributes(f, final_context_object, final_context, original_message)

0 commit comments

Comments
 (0)