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

Adds missing await (#55)

Browse Source 
Co-authored-by: Amir Mousavi <amirh@collisure.com>
This commit is contained in:
Amir Mousavi 2023-05-08 17:46:52 +08:00 committed by GitHub
parent 2d3ac1fe63
commit 115770547c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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 # allow overriding the process method
if self._preprocess is not None and hasattr(self._preprocess, "process"): 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 # Create gRPC stub for communicating with the server
triton_server_address = self._server_config.get("triton_grpc_server") or self._default_grpc_address triton_server_address = self._server_config.get("triton_grpc_server") or self._default_grpc_address