Skip to content

Commit 73f85ca

Browse files
committed
test fix
1 parent ecf7da8 commit 73f85ca

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

datacrunch/InferenceClient/inference_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -324,7 +324,7 @@ def status_json(self) -> Dict[str, Any]:
324324

325325
return response_json
326326

327-
def result(self) -> Dict[str, Any] | str:
327+
def result(self) -> Dict[str, Any]:
328328
"""Get the results of the async inference execution.
329329
330330
Returns:
@@ -337,7 +337,7 @@ def result(self) -> Dict[str, Any] | str:
337337
if response.headers['Content-Type'] == 'application/json':
338338
return response.json()
339339
else:
340-
return response.text
340+
return {'result': response.text}
341341

342342
# alias for get_results
343343
output = result

0 commit comments

Comments
 (0)