fix return type (#2566)

Signed-off-by: jagadeesh <jagadeeshj@ideas2it.com>
pytorch · Sep 5, 2023 · 499d33d · 499d33d
1 parent 16242d6
commit 499d33d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kubernetes/kserve/kserve_wrapper/TorchserveModel.py b/kubernetes/kserve/kserve_wrapper/TorchserveModel.py
@@ -104,7 +104,7 @@ async def _grpc_predict(
         payload = to_ts_grpc(payload)
         grpc_stub = self.grpc_client()
         async_result = await grpc_stub.Predictions(payload)
-        return async_result
+        return from_ts_grpc(async_result)
 
     def postprocess(
         self,