Add optional for exceptions to satisfy pytype (#6495)
diff --git a/bindings/python/iree/runtime/function.py b/bindings/python/iree/runtime/function.py
index 36a5280..e46b29f 100644
--- a/bindings/python/iree/runtime/function.py
+++ b/bindings/python/iree/runtime/function.py
@@ -381,7 +381,9 @@
return dtype(x)
-def _raise_argument_error(inv: Invocation, summary: str, e: Exception = None):
+def _raise_argument_error(inv: Invocation,
+ summary: str,
+ e: Optional[Exception] = None):
new_e = ValueError(f"Error passing argument: {summary} "
f"(while encoding argument {inv.summarize_arg_error()})")
if e:
@@ -390,7 +392,9 @@
raise new_e
-def _raise_return_error(inv: Invocation, summary: str, e: Exception = None):
+def _raise_return_error(inv: Invocation,
+ summary: str,
+ e: Optional[Exception] = None):
new_e = ValueError(f"Error processing function return: {summary} "
f"(while decoding return {inv.summarize_return_error()})")
if e:
