ob-metaflow-extensions 1.1.160__py2.py3-none-any.whl → 1.1.161__py2.py3-none-any.whl

metaflow_extensions/outerbounds/plugins/kubernetes/pod_killer.py

@@ -228,7 +228,7 @@ class PodKiller:
 
         if outcome == JobOutcomes.LEAVE_UNCHANGED:
             # self.echo(f"Job {job_name} is in terminal state, leaving unchanged")
-            return True
+            return None
         elif outcome == JobOutcomes.DELETE:
             self.echo(f"Deleting Job {job_name}")
             return self._delete_job(job_name, namespace)
@@ -251,7 +251,7 @@ class PodKiller:
 
         if outcome == JobOutcomes.LEAVE_UNCHANGED:
             # self.echo(f"JobSet {jobset_name} is in terminal state, leaving unchanged")
-            return True
+            return None
         elif outcome == JobOutcomes.DELETE:
             self.echo(f"Deleting JobSet {jobset_name}")
             try:
@@ -275,13 +275,15 @@ class PodKiller:
         results = []
 
         # Process matching jobs
+        _jobs, _jobsets = [], []
         jobs = self._find_matching_jobs(flow_name, run_id, user)
         for job in jobs:
             outcome = derive_job_outcome(job.status)
             result = self._handle_job_outcome(job, outcome)
-            if not result:
-                self.echo("Failed Result for : %s" % job.metadata.name)
-            results.append(result)
+            # results.append(result)
+            if result is not None:
+                results.append(result)
+                _jobs.append(result)
 
         # Process matching jobsets
         jobsets = self._find_matching_jobsets(flow_name, run_id, user)
@@ -289,8 +291,8 @@ class PodKiller:
             jobset_status = jobset.get("status", {})
             outcome = derive_jobset_outcome(jobset_status)
             result = self._handle_jobset_outcome(jobset, outcome)
-            if not result:
-                self.echo("Failed Result for : %s" % jobset.get("metadata").get("name"))
-            results.append(result)
+            if result is not None:
+                results.append(result)
+                _jobsets.append(result)
 
-        return results, len(jobs), len(jobsets)
+        return results, len(_jobs), len(_jobsets)

{ob_metaflow_extensions-1.1.160.dist-info → ob_metaflow_extensions-1.1.161.dist-info}/METADATA

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

{ob_metaflow_extensions-1.1.160.dist-info → ob_metaflow_extensions-1.1.161.dist-info}/RECORD

@@ -23,7 +23,7 @@ metaflow_extensions/outerbounds/plugins/fast_bakery/fast_bakery_cli.py,sha256=kq
 metaflow_extensions/outerbounds/plugins/fast_bakery/fast_bakery_decorator.py,sha256=MXSIp05-jvt8Q2uGaLKjtuM_ToLeRLxhtMbfHc9Kcko,1515
 metaflow_extensions/outerbounds/plugins/kubernetes/__init__.py,sha256=5zG8gShSj8m7rgF4xgWBZFuY3GDP5n1T0ktjRpGJLHA,69
 metaflow_extensions/outerbounds/plugins/kubernetes/kubernetes_client.py,sha256=sjBhQ4aa-i1UkKsJyTswdDLYOBAFIvHRco4r7wfs9Tc,5003
-metaflow_extensions/outerbounds/plugins/kubernetes/pod_killer.py,sha256=iZQSXNK14DVrJ6jRfrMfkjoUf4dNKDZA51k6LeVJL6M,11392
+metaflow_extensions/outerbounds/plugins/kubernetes/pod_killer.py,sha256=R483_Qg8eRifx5Z-3iHSQsFL6hUDmNTVLBlPWudCsQA,11404
 metaflow_extensions/outerbounds/plugins/nim/card.py,sha256=dXOJvsZed5NyYyxYLPDvtwg9z_X4azL9HTJGYaiNriY,4690
 metaflow_extensions/outerbounds/plugins/nim/nim_decorator.py,sha256=50YVvC7mcZYlPluM0Wq1UtufhzlQb-RxzZkTOJJ3LkM,3439
 metaflow_extensions/outerbounds/plugins/nim/nim_manager.py,sha256=y8U71106KJtrC6nlhsNnzX9Xkv3RnyZ1KEpRFwqZZFk,13686
@@ -73,7 +73,7 @@ metaflow_extensions/outerbounds/toplevel/plugins/kubernetes/__init__.py,sha256=5
 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
 metaflow_extensions/outerbounds/toplevel/plugins/torchtune/__init__.py,sha256=uTVkdSk3xZ7hEKYfdlyVteWj5KeDwaM1hU9WT-_YKfI,50
-ob_metaflow_extensions-1.1.160.dist-info/METADATA,sha256=-Y7aOv3_giiWpXTkncvXFIeHqXqfnWOA0hiHuLqueS8,521
-ob_metaflow_extensions-1.1.160.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
-ob_metaflow_extensions-1.1.160.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
-ob_metaflow_extensions-1.1.160.dist-info/RECORD,,
+ob_metaflow_extensions-1.1.161.dist-info/METADATA,sha256=5iL8FqYFqx1U_GR1YUAzqeV1NrR44jnr_vFSUOmjJEo,521
+ob_metaflow_extensions-1.1.161.dist-info/WHEEL,sha256=bb2Ot9scclHKMOLDEHY6B2sicWOgugjFKaJsT7vwMQo,110
+ob_metaflow_extensions-1.1.161.dist-info/top_level.txt,sha256=NwG0ukwjygtanDETyp_BUdtYtqIA_lOjzFFh1TsnxvI,20
+ob_metaflow_extensions-1.1.161.dist-info/RECORD,,