We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ad09395 commit e9e18a9Copy full SHA for e9e18a9
torch/_export/db/logging.py
@@ -1,7 +1,6 @@
1
-# mypy: allow-untyped-defs
+from typing import Optional
2
3
-
4
-def exportdb_error_message(case_name: str):
+def exportdb_error_message(case_name: str) -> str:
5
from .examples import all_examples
6
from torch._utils_internal import log_export_usage
7
@@ -19,7 +18,7 @@ def exportdb_error_message(case_name: str):
19
18
return f"{case_name} is unsupported."
20
21
22
-def get_class_if_classified_error(e):
+def get_class_if_classified_error(e: Exception) -> Optional[str]:
23
"""
24
Returns a string case name if the export error e is classified.
25
Returns None otherwise.
0 commit comments