Skip to content

Commit f19a236

Browse files
committed
use _build_request_headers method
1 parent a744acb commit f19a236

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
@@ -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

0 commit comments

Comments
 (0)