ClearML/clearml-serving
1
0
Fork 0
mirror of https://github.com/clearml/clearml-serving synced 2025-01-31 02:46:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
allegroai 2023-09-23 17:35:21 +03:00
parent 4a737b95c6
commit e4c07c756a
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
clearml_serving/serving/preprocess_service.py
View File

@ -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)