torchx - add exception_type and exception_message to torchx event

Summary: Add exception type and message to torchx event. This allows for better logging of exception details for further analysis.

Differential Revision: D64406552

Author: Tony Kao

torchx/runner/events/__init__.py

@@ -123,6 +123,10 @@ def __exit__(
         ) // 1000
         if traceback_type:
             self._torchx_event.raw_exception = traceback.format_exc()
+        if exec_type:
+            self._torchx_event.exception_type = exec_type.__name__
+        if exec_value:
+            self._torchx_event.exception_message = str(exec_value)
         record(self._torchx_event)
 
     def _generate_torchx_event(

torchx/runner/events/api.py

@@ -52,6 +52,8 @@ class TorchxEvent:
     wall_time_usec: Optional[int] = None
     start_epoch_time_usec: Optional[int] = None
     workspace: Optional[str] = None
+    exception_type: Optional[str] = None
+    exception_message: Optional[str] = None
 
     def __str__(self) -> str:
         return self.serialize()