From 115770547c0289b763e7de7b246886d68d7720b7 Mon Sep 17 00:00:00 2001 From: Amir Mousavi Date: Mon, 8 May 2023 17:46:52 +0800 Subject: [PATCH] Adds missing await (#55) Co-authored-by: Amir Mousavi --- clearml_serving/serving/preprocess_service.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clearml_serving/serving/preprocess_service.py b/clearml_serving/serving/preprocess_service.py index bbe2306..af4d96d 100644 --- a/clearml_serving/serving/preprocess_service.py +++ b/clearml_serving/serving/preprocess_service.py @@ -316,7 +316,7 @@ class TritonPreprocessRequest(BasePreprocessRequest): """ # allow overriding the process method if self._preprocess is not None and hasattr(self._preprocess, "process"): - return self._preprocess.process(data, state, collect_custom_statistics_fn) + return await self._preprocess.process(data, state, collect_custom_statistics_fn) # Create gRPC stub for communicating with the server triton_server_address = self._server_config.get("triton_grpc_server") or self._default_grpc_address