ob-metaflow-extensions 1.1.53__tar.gz → 1.1.55__tar.gz

{ob-metaflow-extensions-1.1.53 → ob-metaflow-extensions-1.1.55}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-extensions
-Version: 1.1.53
+Version: 1.1.55
 Summary: Outerbounds Platform Extensions for Metaflow
 Author: Outerbounds, Inc.
 License: Commercial

{ob-metaflow-extensions-1.1.53 → ob-metaflow-extensions-1.1.55}/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 → ob-metaflow-extensions-1.1.55}/ob_metaflow_extensions.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-extensions
-Version: 1.1.53
+Version: 1.1.55
 Summary: Outerbounds Platform Extensions for Metaflow
 Author: Outerbounds, Inc.
 License: Commercial

ob-metaflow-extensions-1.1.55/ob_metaflow_extensions.egg-info/requires.txt

@@ -0,0 +1,3 @@
+boto3
+kubernetes
+ob-metaflow==2.11.14.1

{ob-metaflow-extensions-1.1.53 → ob-metaflow-extensions-1.1.55}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
 from pathlib import Path
 
 
-version = "1.1.53"
+version = "1.1.55"
 this_directory = Path(__file__).parent
 long_description = (this_directory / "README.md").read_text()
 
@@ -18,7 +18,6 @@ setup(
     install_requires=[
         "boto3",
         "kubernetes",
-        "ob-metaflow == 2.11.10.3",
-        "ob-metaflow-stubs == 2.11.10.3",
+        "ob-metaflow == 2.11.14.1",
     ],
 )

ob-metaflow-extensions-1.1.53/ob_metaflow_extensions.egg-info/requires.txt

@@ -1,4 +0,0 @@
-boto3
-kubernetes
-ob-metaflow==2.11.10.3
-ob-metaflow-stubs==2.11.10.3