File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
datacrunch/InferenceClient Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -303,7 +303,7 @@ def status(self) -> Dict[str, Any]:
303303 """
304304 url = f'{ self ._inference_client .base_domain } /status/{ self ._inference_client .deployment_name } '
305305 response = self ._inference_client ._session .get (
306- url , headers = {self .INFERENCE_ID_HEADER : self .id , ** self . _inference_client . _global_headers } )
306+ url , headers = self . _inference_client . _build_request_headers ( {self .INFERENCE_ID_HEADER : self .id }) )
307307
308308 response_json = response .json ()
309309 self ._status = response_json ['status' ]
@@ -318,7 +318,7 @@ def result(self) -> Dict[str, Any]:
318318 """
319319 url = f'{ self ._inference_client .base_domain } /results/{ self ._inference_client .deployment_name } '
320320 response = self ._inference_client ._session .get (
321- url , headers = {self .INFERENCE_ID_HEADER : self .id , ** self . _inference_client . _global_headers } )
321+ url , headers = self . _inference_client . _build_request_headers ( {self .INFERENCE_ID_HEADER : self .id }) )
322322
323323 return response
324324 # alias for get_results
You can’t perform that action at this time.
0 commit comments