ob-metaflow-extensions 1.1.145__py2.py3-none-any.whl → 1.1.146__py2.py3-none-any.whl

metaflow_extensions/outerbounds/plugins/nvcf/nvcf.py

@@ -153,7 +153,7 @@ class Nvcf(object):
 
     def wait(self, stdout_location, stderr_location, echo=None):
         def wait_for_launch(job):
-            status = job.status
+            status = job._status
             echo(
                 "Task status: %s..." % status,
                 "stderr",
@@ -215,7 +215,7 @@ class Job(object):
         self._function_id = function_id
         self._ngc_api_key = ngc_api_key
         self._queue_timeout = queue_timeout
-        self._poll_seconds = "300"
+        self._poll_seconds = "3600"
 
         # Initialize status and tracking variables
         self._status = JobStatus.CREATED
@@ -299,7 +299,7 @@ class Job(object):
             polling_thread.start()
 
     def _long_poll_loop(self):
-        while self._long_polling_active and self.status not in terminal_states:
+        while self._long_polling_active and self._status not in terminal_states:
             try:
                 self._poll()
                 # No sleep needed - the request itself will block for up to self._poll_seconds
@@ -310,16 +310,6 @@ class Job(object):
 
         self._long_polling_active = False
 
-    @property
-    def status(self):
-        # If status is already terminal, don't poll again
-        if self._status in terminal_states:
-            return self._status
-
-        # Return cached status - no need to poll
-        # Long polling loop will update the status
-        return self._status
-
     @property
     def id(self):
         return self._invocation_id
@@ -327,11 +317,11 @@ class Job(object):
     @property
     def is_running(self):
         # Job is running if it's in SUBMITTED or POLLED state
-        return self.status in [JobStatus.SUBMITTED, JobStatus.POLLED]
+        return self._status in [JobStatus.SUBMITTED, JobStatus.POLLED]
 
     @property
     def has_failed(self):
-        return self.status == JobStatus.FAILED
+        return self._status == JobStatus.FAILED
 
     @property
     def result(self):
@@ -416,6 +406,9 @@ class Job(object):
                     )
                     self._status = JobStatus.FAILED
                     raise NvcfPollingConnectionError(e)
+            elif e.code in [500, 504]:
+                # Don't set status to FAILED, just re-raise for retry decorator
+                raise
             else:
                 self._status = JobStatus.FAILED
                 raise NvcfPollingConnectionError(e)

{ob_metaflow_extensions-1.1.145.dist-info → ob_metaflow_extensions-1.1.146.dist-info}/METADATA

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

{ob_metaflow_extensions-1.1.145.dist-info → ob_metaflow_extensions-1.1.146.dist-info}/RECORD

@@ -28,7 +28,7 @@ metaflow_extensions/outerbounds/plugins/nvcf/__init__.py,sha256=47DEQpj8HBSa-_TI
 metaflow_extensions/outerbounds/plugins/nvcf/constants.py,sha256=aGHdNw_hqBu8i0zWXcatQM6e769wUXox0l8g0f6fNZ8,146
 metaflow_extensions/outerbounds/plugins/nvcf/exceptions.py,sha256=-Pm9cOWUzpv94TvVUeq-FenAWdfLBJd5N7WPqIGZVqU,3671
 metaflow_extensions/outerbounds/plugins/nvcf/heartbeat_store.py,sha256=pOWwm8LFQBbtku0zNBBwCyXxLK8U-hhC4naQcmU69nE,6217
-metaflow_extensions/outerbounds/plugins/nvcf/nvcf.py,sha256=PVX7rrpP_C63f_ZK8V0PXBpFYBMgyKcxRi8BEe5Dq3g,15745
+metaflow_extensions/outerbounds/plugins/nvcf/nvcf.py,sha256=3ZFdYItVpFWnHMOeyV1nslUyelfvX5rknh2d2IWxVws,15591
 metaflow_extensions/outerbounds/plugins/nvcf/nvcf_cli.py,sha256=3D-r5XO88Yh2k1EAZFJTe_PwdbhWp5qXflG8AgE4ZIU,9500
 metaflow_extensions/outerbounds/plugins/nvcf/nvcf_decorator.py,sha256=pxxNxW4bW3kbB6ybRam657GyKIhvIkMuidB94iFfCD8,9116
 metaflow_extensions/outerbounds/plugins/nvcf/utils.py,sha256=DxWSCayfa95e0HJkWacey1s1nxoTpaunGhrb_0Ayv28,133
@@ -58,7 +58,7 @@ metaflow_extensions/outerbounds/toplevel/plugins/gcp/__init__.py,sha256=BbZiaH3u
 metaflow_extensions/outerbounds/toplevel/plugins/kubernetes/__init__.py,sha256=5zG8gShSj8m7rgF4xgWBZFuY3GDP5n1T0ktjRpGJLHA,69
 metaflow_extensions/outerbounds/toplevel/plugins/ollama/__init__.py,sha256=GRSz2zwqkvlmFS6bcfYD_CX6CMko9DHQokMaH1iBshA,47
 metaflow_extensions/outerbounds/toplevel/plugins/snowflake/__init__.py,sha256=LptpH-ziXHrednMYUjIaosS1SXD3sOtF_9_eRqd8SJw,50
-ob_metaflow_extensions-1.1.145.dist-info/METADATA,sha256=OBVl7pDNOMlA8lmGSi60VRYK_4_TD36y577huIfwpu4,520
-ob_metaflow_extensions-1.1.145.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-ob_metaflow_extensions-1.1.145.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
-ob_metaflow_extensions-1.1.145.dist-info/RECORD,,
+ob_metaflow_extensions-1.1.146.dist-info/METADATA,sha256=Vtp-8yJaPeg8Hh33WXQYAerLoLlx5Ldya5M2U1lp5z0,520
+ob_metaflow_extensions-1.1.146.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
+ob_metaflow_extensions-1.1.146.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
+ob_metaflow_extensions-1.1.146.dist-info/RECORD,,