add exec info for warnings

Signed-off-by: Kelly A <kellyaa@users.noreply.github.com>
kellyaa · May 14, 2024 · 4032a0c · 4032a0c
1 parent d4c4a08
commit 4032a0c
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/caikit/runtime/train.py b/caikit/runtime/train.py
@@ -155,7 +155,8 @@ def main() -> int:
                 "log_code": "<COR39662029E>",
                 "message": message,
                 "stack_trace": traceback.format_exc(),
-            }
+            },
+            exc_info=True,
         )
         write_termination_log(message)
         exit(USER_ERROR_EXIT_CODE)
@@ -172,7 +173,8 @@ def main() -> int:
                 "log_code": "<COR17776539E>",
                 "message": message,
                 "stack_trace": traceback.format_exc(),
-            }
+            },
+            exc_info=True,
         )
         write_termination_log(message)
         exit(USER_ERROR_EXIT_CODE)
@@ -269,7 +271,8 @@ def main() -> int:
                 "log_code": "<COR65474760E>",
                 "message": message,
                 "stack_trace": traceback.format_exc(),
-            }
+            },
+            exc_info=True,
         )
     except Exception:
         message = "Exception encountered when attempting to parse input parameters"
@@ -278,7 +281,8 @@ def main() -> int:
                 "log_code": "<COR17776549E>",
                 "message": message,
                 "stack_trace": traceback.format_exc(),
-            }
+            },
+            exc_info=True,
         )
         write_termination_log(message)
         exit(USER_ERROR_EXIT_CODE)
@@ -308,7 +312,8 @@ def main() -> int:
                 "log_code": "<COR04280062E>",
                 "message": message,
                 "stack_trace": traceback.format_exc(),
-            }
+            },
+            exc_info=True,
         )
         write_termination_log(message)
         exit(INTERNAL_ERROR_EXIT_CODE)
@@ -319,7 +324,8 @@ def main() -> int:
                 "log_code": "<COR04280062E>",
                 "message": message,
                 "stack_trace": traceback.format_exc(),
-            }
+            },
+            exc_info=True,
         )
         write_termination_log(message)
         exit(INTERNAL_ERROR_EXIT_CODE)