ob-metaflow-extensions 1.1.72__py2.py3-none-any.whl → 1.1.73__py2.py3-none-any.whl

metaflow_extensions/outerbounds/plugins/auth_server.py

@@ -52,10 +52,17 @@ def get_token(url_path):
             return token_info
         except requests.exceptions.HTTPError as e:
             raise MetaflowException(repr(e))
-        except requests.exceptions.ConnectionError as e:
+        except (
+            requests.exceptions.ConnectionError,
+            requests.exceptions.ReadTimeout,
+        ) as e:
             # ConnectionErrors are generally temporary errors like DNS resolution failures,
             # timeouts etc.
-            print("received connection error. Retrying...", e, file=sys.stderr)
+            print(
+                "received error of type {}. Retrying...".format(type(e)),
+                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

metaflow_extensions/outerbounds/plugins/nim/nim_manager.py

@@ -174,11 +174,16 @@ class NimChatCompletion(object):
                         self.invocations.append(invocation_id)
                     elif response.status_code == 200:
                         return response.json()
-                except requests.exceptions.ConnectionError as e:
+                except (
+                    requests.exceptions.ConnectionError,
+                    requests.exceptions.ReadTimeout,
+                ) as e:
                     # ConnectionErrors are generally temporary errors like DNS resolution failures,
                     # timeouts etc.
                     print(
-                        "Encountered connection error. Retrying...", e, file=sys.stderr
+                        "received error of type {}. Retrying...".format(type(e)),
+                        e,
+                        file=sys.stderr,
                     )
                     time.sleep(retry_delay)
                     retry_delay *= 2  # Double the delay for the next attempt
@@ -205,11 +210,14 @@ class NimChatCompletion(object):
                             raise Exception(
                                 f"NVCF returned {poll_response.status_code} status code. Please contact Outerbounds."
                             )
-                    except requests.exceptions.ConnectionError as e:
+                    except (
+                        requests.exceptions.ConnectionError,
+                        requests.exceptions.ReadTimeout,
+                    ) as e:
                         # ConnectionErrors are generally temporary errors like DNS resolution failures,
                         # timeouts etc.
                         print(
-                            "Encountered connection error. Retrying...",
+                            "received error of type {}. Retrying...".format(type(e)),
                             e,
                             file=sys.stderr,
                         )

{ob_metaflow_extensions-1.1.72.dist-info → ob_metaflow_extensions-1.1.73.dist-info}/METADATA

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

{ob_metaflow_extensions-1.1.72.dist-info → ob_metaflow_extensions-1.1.73.dist-info}/RECORD

@@ -2,12 +2,12 @@ metaflow_extensions/outerbounds/__init__.py,sha256=TRGvIUMjkfneWtYUFSWoubu_Kf2ek
 metaflow_extensions/outerbounds/remote_config.py,sha256=Zpfpjgz68_ZgxlXezjzlsDLo4840rkWuZgwDB_5H57U,4059
 metaflow_extensions/outerbounds/config/__init__.py,sha256=mYo95obHU1IE1wbPkeVz_pfTzNqlNabp1QBEMTGllbE,112
 metaflow_extensions/outerbounds/plugins/__init__.py,sha256=46NgbJBhVowDR6FyQrZPF2jHHqRTSyCBCYIQAyQ4Ryo,9516
-metaflow_extensions/outerbounds/plugins/auth_server.py,sha256=JhlMFcR7SPSfR1C9w6GlqJq-NYNhOfISmHl2PdkYUok,2212
+metaflow_extensions/outerbounds/plugins/auth_server.py,sha256=1v2GBqoMBxp5E7Lejz139w-jxJtPnLDvvHXP0HhEIHI,2361
 metaflow_extensions/outerbounds/plugins/perimeters.py,sha256=QXh3SFP7GQbS-RAIxUOPbhPzQ7KDFVxZkTdKqFKgXjI,2697
 metaflow_extensions/outerbounds/plugins/kubernetes/__init__.py,sha256=5zG8gShSj8m7rgF4xgWBZFuY3GDP5n1T0ktjRpGJLHA,69
 metaflow_extensions/outerbounds/plugins/kubernetes/kubernetes_client.py,sha256=gj6Iaz26bGbZm3aQuNS18Mqh_80iJp5PgFwFSlJRcn8,1968
 metaflow_extensions/outerbounds/plugins/nim/__init__.py,sha256=GVnvSTjqYVj5oG2yh8KJFt7iZ33cEadDD5HbdmC9hJ0,1457
-metaflow_extensions/outerbounds/plugins/nim/nim_manager.py,sha256=XAI9WVmRGVG0ZvK9lvVNLHkh5qYgMJbl9VOAARJrT3g,8931
+metaflow_extensions/outerbounds/plugins/nim/nim_manager.py,sha256=SWieODDxtIaeZwdMYtObDi57Kjyfw2DUuE6pJtU750w,9206
 metaflow_extensions/outerbounds/plugins/nvcf/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 metaflow_extensions/outerbounds/plugins/nvcf/nvcf.py,sha256=ftxC5SCo64P5Ycpv5vudluTnQi3-VCZW0umdsPP326A,7926
 metaflow_extensions/outerbounds/plugins/nvcf/nvcf_cli.py,sha256=ow3lonclEDoZEUQCDV_L8lEr6HopXqjNXzubRrfdIm4,7219
@@ -19,7 +19,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.72.dist-info/METADATA,sha256=B9Trsr_O9Q_ZDZZ5a8gZoAoD5egZKgpUnURzZJIYxzo,519
-ob_metaflow_extensions-1.1.72.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-ob_metaflow_extensions-1.1.72.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
-ob_metaflow_extensions-1.1.72.dist-info/RECORD,,
+ob_metaflow_extensions-1.1.73.dist-info/METADATA,sha256=d45Pkzj0dMiznAUc91auqJVpgAp7o7QD_R5DHIKqwgg,519
+ob_metaflow_extensions-1.1.73.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
+ob_metaflow_extensions-1.1.73.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
+ob_metaflow_extensions-1.1.73.dist-info/RECORD,,