Skip to content

Commit 2acb246

Browse files
authored
Pass raise_on_error to response iterators (#439)
1 parent 109c69f commit 2acb246

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/tritonserver/_api/_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,7 @@ def async_infer(
179179

180180
request = inference_request._create_tritonserver_inference_request()
181181
self._server.infer_async(request)
182-
return AsyncResponseIterator(self, request, inference_request)
182+
return AsyncResponseIterator(self, request, inference_request, raise_on_error)
183183

184184
def infer(
185185
self,
@@ -250,7 +250,7 @@ def infer(
250250

251251
request = inference_request._create_tritonserver_inference_request()
252252
self._server.infer_async(request)
253-
return ResponseIterator(self, request, inference_request)
253+
return ResponseIterator(self, request, inference_request, raise_on_error)
254254

255255
def metadata(self) -> dict[str, Any]:
256256
"""Returns medatadata about a model and its inputs and outputs

0 commit comments

Comments
 (0)