Add traceback for failing to load preprocess class (#57)

clearml · Sep 23, 2023 · e4c07c7 · e4c07c7
1 parent 4a737b9
commit e4c07c7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/clearml_serving/serving/preprocess_service.py b/clearml_serving/serving/preprocess_service.py
@@ -1,6 +1,7 @@
 import os
 import sys
 import threading
+import traceback
 from pathlib import Path
 from typing import Optional, Any, Callable, List
 
@@ -48,8 +49,8 @@ def __init__(
                 try:
                     self._instantiate_custom_preprocess_cls(task)
                 except Exception as ex:
-                    raise ValueError("Error: Failed loading preprocess code for \'{}\': {}".format(
-                        self.model_endpoint.preprocess_artifact, ex))
+                    raise ValueError("Error: Failed loading preprocess code for \'{}\': {}\n\n{}".format(
+                        self.model_endpoint.preprocess_artifact, ex, traceback.format_exc()))
 
     def _instantiate_custom_preprocess_cls(self, task: Task) -> None:
         path = task.artifacts[self.model_endpoint.preprocess_artifact].get_local_copy(extract_archive=False)