increase status_code check to >= 500 (#73)

Author: jonathanfeng-scale

scaleapi/_version.py

@@ -1,2 +1,2 @@
-__version__ = "2.14.0"
+__version__ = "2.14.1"
 __package_name__ = "scaleapi"

scaleapi/api.py

@@ -152,8 +152,8 @@ def _api_request(
             #   status=409,
             #   redirect_location=None)
             if retry_history != ():
-                # See if the first retry was a 500 error
-                if retry_history[0][3] == 500:
+                # See if the first retry was a 500 or 503 error
+                if retry_history[0][3] >= 500:
                     uuid = body["unique_id"]
                     newUrl = f"{self.base_api_url}/tasks?unique_id={uuid}"
                     # grab task from api