ob-metaflow-extensions 1.1.53__py2.py3-none-any.whl → 1.1.54__py2.py3-none-any.whl

metaflow_extensions/outerbounds/plugins/auth_server.py

@@ -1,6 +1,10 @@
 from urllib.parse import urlparse
 
 import requests
+import time
+import random
+import os
+import sys
 
 
 def read_mf_config():
@@ -35,12 +39,26 @@ def get_token_url_and_headers(url_path):
 def get_token(url_path):
     from metaflow.exception import MetaflowException
 
-    url, headers = get_token_url_and_headers(url_path)
-    try:
-        r = requests.get(url, headers=headers)
-        r.raise_for_status()
-        token_info = r.json()
-        return token_info
+    retry_delay = 1
+    max_retries = os.environ.get("METAFLOW_EXT_HTTP_MAX_RETRIES", "10")
+    attempt = 0
+    while attempt < int(max_retries):
+        attempt = attempt + 1
+        try:
+            url, headers = get_token_url_and_headers(url_path)
+            r = requests.get(url, headers=headers)
+            r.raise_for_status()
+            token_info = r.json()
+            return token_info
+        except requests.exceptions.HTTPError as e:
+            raise MetaflowException(repr(e))
+        except requests.exceptions.ConnectionError as e:
+            # ConnectionErrors are generally temporary errors like DNS resolution failures,
+            # timeouts etc.
+            print("received connection error. Retrying...", e, file=sys.stderr)
+            time.sleep(retry_delay)
+            retry_delay *= 2  # Double the delay for the next attempt
+            retry_delay += random.uniform(0, 1)  # Add jitter
+            retry_delay = min(retry_delay, 10)
 
-    except requests.exceptions.HTTPError as e:
-        raise MetaflowException(repr(e))
+    raise MetaflowException("exceeded retries from get_token")

{ob_metaflow_extensions-1.1.53.dist-info → ob_metaflow_extensions-1.1.54.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-extensions
-Version: 1.1.53
+Version: 1.1.54
 Summary: Outerbounds Platform Extensions for Metaflow
 Author: Outerbounds, Inc.
 License: Commercial
@@ -8,7 +8,6 @@ Description-Content-Type: text/markdown
 Requires-Dist: boto3
 Requires-Dist: kubernetes
 Requires-Dist: ob-metaflow (==2.11.10.3)
-Requires-Dist: ob-metaflow-stubs (==2.11.10.3)
 
 # Outerbounds platform package
 

{ob_metaflow_extensions-1.1.53.dist-info → ob_metaflow_extensions-1.1.54.dist-info}/RECORD

@@ -2,7 +2,7 @@ metaflow_extensions/outerbounds/__init__.py,sha256=TRGvIUMjkfneWtYUFSWoubu_Kf2ek
 metaflow_extensions/outerbounds/remote_config.py,sha256=jHNHDbWAoKOgv3IH2I-Gzwfe2tr1gO3ds_VjsXBaCh8,4045
 metaflow_extensions/outerbounds/config/__init__.py,sha256=mYo95obHU1IE1wbPkeVz_pfTzNqlNabp1QBEMTGllbE,112
 metaflow_extensions/outerbounds/plugins/__init__.py,sha256=rK-EPg7107su6dY-yAx1IK3wAzyvQIIg6y4um_F_BXc,9343
-metaflow_extensions/outerbounds/plugins/auth_server.py,sha256=rUzhRfE0ov3pWTM9S4UdAhnz13JBb5amiQCs5d1xJQU,1418
+metaflow_extensions/outerbounds/plugins/auth_server.py,sha256=JhlMFcR7SPSfR1C9w6GlqJq-NYNhOfISmHl2PdkYUok,2212
 metaflow_extensions/outerbounds/plugins/perimeters.py,sha256=z8tSAkWtiITB-JtSQS7fkhlBwvxSxeTgEwFjahAzv-U,2238
 metaflow_extensions/outerbounds/plugins/kubernetes/__init__.py,sha256=5zG8gShSj8m7rgF4xgWBZFuY3GDP5n1T0ktjRpGJLHA,69
 metaflow_extensions/outerbounds/plugins/kubernetes/kubernetes_client.py,sha256=gj6Iaz26bGbZm3aQuNS18Mqh_80iJp5PgFwFSlJRcn8,1968
@@ -13,7 +13,7 @@ metaflow_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.py,
 metaflow_extensions/outerbounds/toplevel/plugins/azure/__init__.py,sha256=WUuhz2YQfI4fz7nIcipwwWq781eaoHEk7n4GAn1npDg,63
 metaflow_extensions/outerbounds/toplevel/plugins/gcp/__init__.py,sha256=BbZiaH3uILlEZ6ntBLKeNyqn3If8nIXZFq_Apd7Dhco,70
 metaflow_extensions/outerbounds/toplevel/plugins/kubernetes/__init__.py,sha256=5zG8gShSj8m7rgF4xgWBZFuY3GDP5n1T0ktjRpGJLHA,69
-ob_metaflow_extensions-1.1.53.dist-info/METADATA,sha256=xkhC6JdJnBYOhW_pcDWbr5v6lZLw2h5JucLUtjTrTX4,567
-ob_metaflow_extensions-1.1.53.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-ob_metaflow_extensions-1.1.53.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
-ob_metaflow_extensions-1.1.53.dist-info/RECORD,,
+ob_metaflow_extensions-1.1.54.dist-info/METADATA,sha256=dGtJ8ND8Nyb7lcKSjAFbGbQN-nYBKSYod5zWwW0TviE,520
+ob_metaflow_extensions-1.1.54.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
+ob_metaflow_extensions-1.1.54.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
+ob_metaflow_extensions-1.1.54.dist-info/RECORD,,