Fix backoff error for max_tries argument

LeMyst · Sep 6, 2024 · 66ab27b · 66ab27b
1 parent 22d0e53
commit 66ab27b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/wikibaseintegrator/wbi_backoff.py b/wikibaseintegrator/wbi_backoff.py
@@ -28,8 +28,10 @@ def wbi_backoff_check_json_decode_error(e) -> bool:
     """
     return isinstance(e, JSONDecodeError) and str(e) != "Expecting value: line 1 column 1 (char 0)"
 
+def wbi_get_backoff_max_tries():
+    return config.get('BACKOFF_MAX_TRIES')
 
 wbi_backoff_exceptions = (requests.exceptions.Timeout, requests.exceptions.ConnectionError, requests.HTTPError, JSONDecodeError)
 
 wbi_backoff = partial(backoff.on_exception, backoff.expo, wbi_backoff_exceptions, max_value=partial(config.get, 'BACKOFF_MAX_VALUE'), giveup=wbi_backoff_check_json_decode_error,
-                      on_backoff=wbi_backoff_backoff_hdlr, jitter=None, max_tries=partial(config.get, 'BACKOFF_MAX_TRIES'))
+                      on_backoff=wbi_backoff_backoff_hdlr, jitter=None, max_tries=wbi_get_backoff_max_tries)