From 4032a0c890de39cc9a9f53cc8a2fd2e0cfaeb26d Mon Sep 17 00:00:00 2001
From: Kelly A <kellyaa@users.noreply.github.com>
Date: Tue, 14 May 2024 17:04:39 -0400
Subject: [PATCH] add exec info for warnings

Signed-off-by: Kelly A <kellyaa@users.noreply.github.com>
---
 caikit/runtime/train.py | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/caikit/runtime/train.py b/caikit/runtime/train.py
index e09a9017c..fe26990cb 100644
--- 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)