feat: add support for 504 error

danielnsilva · Jan 23, 2023 · ec23182 · ec23182
1 parent 1d74d22
commit ec23182
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/semanticscholar/ApiRequester.py b/semanticscholar/ApiRequester.py
@@ -78,7 +78,7 @@ def get_data(
             raise ObjectNotFoundExeception(data['error'])
         elif r.status_code == 429:
             raise ConnectionRefusedError('HTTP status 429 Too Many Requests.')
-        elif r.status_code == 500:
+        elif r.status_code in [500, 504]:
             data = r.json()
             raise Exception(data['message'])