From e435dc25d2f441a06aaf61a8fe17ff3f12cec135 Mon Sep 17 00:00:00 2001 From: Fatih Kurtoglu Date: Tue, 4 Jun 2024 11:40:44 -0700 Subject: [PATCH] Handle non-existing retries within Session object --- scaleapi/_version.py | 2 +- scaleapi/api.py | 11 +++++++---- 2 files changed, 8 insertions(+), 5 deletions(-) diff --git a/scaleapi/_version.py b/scaleapi/_version.py index 96ef07b..f4a4da0 100644 --- a/scaleapi/_version.py +++ b/scaleapi/_version.py @@ -1,2 +1,2 @@ -__version__ = "2.15.6" +__version__ = "2.15.7" __package_name__ = "scaleapi" diff --git a/scaleapi/api.py b/scaleapi/api.py index 43083e1..58cbf19 100644 --- a/scaleapi/api.py +++ b/scaleapi/api.py @@ -144,7 +144,10 @@ def _api_request( # Some endpoints only return 'OK' message without JSON return json elif res.status_code == 409 and "task" in endpoint and body.get("unique_id"): - retry_history = res.raw.retries.history + retry_history = None + if "retries" in res.raw and "history" in res.raw.retries: + retry_history = res.raw.retries.history + # Example RequestHistory tuple # RequestHistory(method='POST', # url='/v1/task/imageannotation', @@ -184,9 +187,9 @@ def post_request(self, endpoint, body=None, files=None, data=None): return self._api_request( "POST", endpoint, - headers=self._headers - if files is None - else self._headers_multipart_form_data, + headers=( + self._headers if files is None else self._headers_multipart_form_data + ), auth=self._auth, body=body, files=files,