From ec2318232f2dee49b1bc9a95a0c6281a97efac0c Mon Sep 17 00:00:00 2001 From: Daniel Silva Date: Mon, 23 Jan 2023 14:49:23 -0500 Subject: [PATCH] feat: add support for 504 error --- semanticscholar/ApiRequester.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/semanticscholar/ApiRequester.py b/semanticscholar/ApiRequester.py index 413839d..2aef3b1 100644 --- 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'])