Adds missing await (#55)

Co-authored-by: Amir Mousavi <[email protected]>
clearml · May 8, 2023 · 1157705 · 1157705
1 parent 2d3ac1f
commit 1157705
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/clearml_serving/serving/preprocess_service.py b/clearml_serving/serving/preprocess_service.py
@@ -316,7 +316,7 @@ async def process(
         """
         # 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