ClearML/clearml-serving
1
0
Fork 0
mirror of https://github.com/clearml/clearml-serving synced 2025-06-26 18:16:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Adds missing await

Browse Source
This commit is contained in:
Amir Mousavi 2023-05-08 12:30:45 +08:00
parent 2d3ac1fe63
commit 7186863997
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clearml_serving/serving/preprocess_service.py
View File

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