zenml-nightly 0.84.0.dev20250721__py3-none-any.whl → 0.84.0.dev20250722__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.84.0.dev20250721
+0.84.0.dev20250722

zenml/integrations/kubernetes/orchestrators/kube_utils.py

@@ -33,6 +33,7 @@ Adjusted from https://github.com/tensorflow/tfx/blob/master/tfx/utils/kube_utils
 
 import enum
 import functools
+import json
 import re
 import time
 from collections import defaultdict
@@ -516,9 +517,19 @@ def create_and_wait_for_pod_to_start(
             retries += 1
             if retries < startup_max_retries:
                 logger.debug(f"The {pod_display_name} failed to start: {e}")
+                message = ""
+                try:
+                    if isinstance(e, ApiException) and e.body:
+                        exception_body = json.loads(e.body)
+                        message = exception_body.get("message", "")
+                except Exception:
+                    pass
                 logger.error(
                     f"Failed to create {pod_display_name}. "
                     f"Retrying in {startup_failure_delay} seconds..."
+                    "\nReason: " + message
+                    if message
+                    else ""
                 )
                 time.sleep(startup_failure_delay)
                 startup_failure_delay *= startup_failure_backoff

zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py

@@ -372,7 +372,29 @@ def main() -> None:
 
                 logger.info(f"Job for step `{step_name}` completed.")
             except Exception:
-                logger.error(f"Job for step `{step_name}` failed.")
+                reason = ""
+                try:
+                    pods_by_job = core_api.list_namespaced_pod(
+                        label_selector=f"job-name={job_name}",
+                        namespace=namespace,
+                    ).items
+                    first_pod = pods_by_job[0]
+                    if status := first_pod.status:
+                        if container_statuses := status.container_statuses:
+                            for cs in container_statuses:
+                                if cs.name == "main":
+                                    if state := cs.state:
+                                        if terminated := state.terminated:
+                                            if terminated.exit_code != 0:
+                                                reason = f"{terminated.reason}(exit_code={terminated.exit_code})"
+                                    break
+                except Exception:
+                    pass
+                logger.error(
+                    f"Job for step `{step_name}` failed. Reason: {reason}"
+                    if reason
+                    else ""
+                )
 
                 raise
 

{zenml_nightly-0.84.0.dev20250721.dist-info → zenml_nightly-0.84.0.dev20250722.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.84.0.dev20250721
+Version: 0.84.0.dev20250722
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops

{zenml_nightly-0.84.0.dev20250721.dist-info → zenml_nightly-0.84.0.dev20250722.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=1BKBh3mVw4_ENdMVlPVblZPtprZV3pHejAw2IFZvBh0,19
+zenml/VERSION,sha256=N0mLZwHSE8u7wmEm8Cz4CJ72SbbKxZGLIiBPX8lXZmw,19
 zenml/__init__.py,sha256=r7JUg2SVDf_dPhS7iU6vudKusEqK4ics7_jFMZhq0o4,2731
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -337,9 +337,9 @@ zenml/integrations/kubernetes/flavors/__init__.py,sha256=a5gU45qCj3FkLwl_uVjlIkL
 zenml/integrations/kubernetes/flavors/kubernetes_orchestrator_flavor.py,sha256=MqPOFJpxQ1hfzF3c_FU_431OJGHOVKT_VNwdAzNu1lc,13595
 zenml/integrations/kubernetes/flavors/kubernetes_step_operator_flavor.py,sha256=Vp7kY9t2TWdWRBNSf_sJKPL5dZnhuHnskPhQ7KBuGPY,6384
 zenml/integrations/kubernetes/orchestrators/__init__.py,sha256=TJID3OTieZBox36WpQpzD0jdVRA_aZVcs_bNtfXS8ik,811
-zenml/integrations/kubernetes/orchestrators/kube_utils.py,sha256=ReqiXBfmtdmFcRU0WIOPfkZO57n-AFGdo-WweNr60VQ,24698
+zenml/integrations/kubernetes/orchestrators/kube_utils.py,sha256=9oPdp1jREPOn2NAu_mmdxuFeYjMartj3cWYqzb808qw,25112
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py,sha256=Agx_Thj6nByAd5p61HTf0p6SRDPOcYKSSl-_Km2EdfU,41316
-zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py,sha256=9swZf_e_rmtQZyQoBMzaweV_w9Tp_iBCmShJBHuveSM,19605
+zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py,sha256=8XZZMJB8E4HnXSUC436VRKMRJHCI57YK7a892xYGTSo,20675
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint_configuration.py,sha256=QOwQnWCfB-t_BQ2eOZN0SakurGUd0GTMCSdUlREhk6I,2324
 zenml/integrations/kubernetes/orchestrators/manifest_utils.py,sha256=52Ba6MZ4lCo6CpnW0IjnFOJQNpPoVgbvIqR-x32ubew,16085
 zenml/integrations/kubernetes/pod_settings.py,sha256=s6I0sALXB7PnRqJ-nlU71o3cTm2GM1TfqsE44qhiDJY,8022
@@ -1347,8 +1347,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=LPFW757WCJLP1S8vrvjs
 zenml/zen_stores/sql_zen_store.py,sha256=cXcKBBRKiwM2GtwHzmJb6tiN5NhGWo9n8SWnL1b4_WE,491150
 zenml/zen_stores/template_utils.py,sha256=iCXrXpqzVTY7roqop4Eh9J7DmLW6PQeILZexmw_l3b8,10074
 zenml/zen_stores/zen_store_interface.py,sha256=weiSULdI9AsbCE10a5TcwtybX-BJs9hKhjPJnTapWv4,93023
-zenml_nightly-0.84.0.dev20250721.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.84.0.dev20250721.dist-info/METADATA,sha256=YRBITfYBSmJb3QkA3nuWle72EFBdFlRO-fBFpJCQBxA,24296
-zenml_nightly-0.84.0.dev20250721.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.84.0.dev20250721.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.84.0.dev20250721.dist-info/RECORD,,
+zenml_nightly-0.84.0.dev20250722.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.84.0.dev20250722.dist-info/METADATA,sha256=e0qGFoH9oo-DHWg46xF-M2ULyoLimghGwAiPkGynU_g,24296
+zenml_nightly-0.84.0.dev20250722.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.84.0.dev20250722.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.84.0.dev20250722.dist-info/RECORD,,