From e4c07c756ad1d5992c272c0bd7a302001dc5f6f2 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Sat, 23 Sep 2023 17:35:21 +0300
Subject: [PATCH] Add traceback for failing to load preprocess class (#57)

---
 clearml_serving/serving/preprocess_service.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/clearml_serving/serving/preprocess_service.py b/clearml_serving/serving/preprocess_service.py
index af4d96d..8e244c0 100644
--- 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 @@ class BasePreprocessRequest(object):
                 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)