Print traceback on model prediction exception.

chassisml · Nov 17, 2023 · 2932bb0 · 2932bb0
1 parent ceca396
commit 2932bb0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/packages/chassisml/src/chassis/runtime/model_runner.py b/packages/chassisml/src/chassis/runtime/model_runner.py
@@ -2,6 +2,8 @@
 
 import os
 import json
+import traceback
+
 import cloudpickle
 from typing import List, Mapping, Optional, Sequence, cast
 
@@ -106,6 +108,7 @@ def _predict_single(self, inputs: Sequence[Mapping[str, bytes]]) -> Sequence[Map
                 output: Mapping[str, bytes] = predict_fn(input_item)
             except Exception as e:
                 print(f"Error: {e}")
+                traceback.print_exc()
                 # TODO - is there more information we can include here like a backtrace?
                 # TODO - convert the error to bytes
                 output = {"error": f"{e}".encode()}