dagster-k8s 0.23.11__tar.gz → 0.23.13__tar.gz

{dagster-k8s-0.23.11/dagster_k8s.egg-info → dagster-k8s-0.23.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-k8s
-Version: 0.23.11
+Version: 0.23.13
 Summary: A Dagster integration for k8s
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-k8s
 Author: Dagster Labs

{dagster-k8s-0.23.11 → dagster-k8s-0.23.13}/dagster_k8s/models.py

@@ -5,7 +5,7 @@ from typing import Any, Mapping, Type, TypeVar
 import dagster._check as check
 import kubernetes
 import kubernetes.client.models
-from dateutil.parser import parse
+from dagster._vendored.dateutil.parser import parse
 from kubernetes.client.api_client import ApiClient
 
 # Unclear what the correct type is to use for a bound here.

{dagster-k8s-0.23.11 → dagster-k8s-0.23.13}/dagster_k8s/ops/k8s_job_op.py

@@ -1,3 +1,4 @@
+import os
 import time
 from typing import Any, Dict, List, Optional
 
@@ -18,7 +19,7 @@ from dagster._annotations import experimental
 from dagster._core.errors import DagsterExecutionInterruptedError
 from dagster._utils.merger import merge_dicts
 
-from ..client import DEFAULT_JOB_POD_COUNT, DagsterKubernetesClient
+from ..client import DEFAULT_JOB_POD_COUNT, DagsterKubernetesClient, k8s_api_retry
 from ..container_context import K8sContainerContext
 from ..job import (
     DagsterK8sJobConfig,
@@ -377,9 +378,18 @@ def execute_k8s_job(
                 if timeout and time.time() - start_time > timeout:
                     watch.stop()
                     raise Exception("Timed out waiting for pod to finish")
-
                 try:
-                    log_entry = next(log_stream)
+                    log_entry = k8s_api_retry(
+                        lambda: next(log_stream),
+                        max_retries=int(
+                            os.getenv("DAGSTER_EXECUTE_K8S_JOB_STREAM_LOGS_RETRIES", "3")
+                        ),
+                        timeout=int(
+                            os.getenv(
+                                "DAGSTER_EXECUTE_K8S_JOB_STREAM_LOGS_WAIT_BETWEEN_ATTEMPTS", "5"
+                            )
+                        ),
+                    )
                     print(log_entry)  # noqa: T201
                 except StopIteration:
                     break

dagster-k8s-0.23.13/dagster_k8s/version.py

@@ -0,0 +1 @@
+__version__ = "0.23.13"

{dagster-k8s-0.23.11 → dagster-k8s-0.23.13/dagster_k8s.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-k8s
-Version: 0.23.11
+Version: 0.23.13
 Summary: A Dagster integration for k8s
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-k8s
 Author: Dagster Labs

{dagster-k8s-0.23.11 → dagster-k8s-0.23.13}/dagster_k8s.egg-info/requires.txt

@@ -1,3 +1,3 @@
-dagster==1.7.11
+dagster==1.7.13
 kubernetes
 google-auth!=2.23.1

{dagster-k8s-0.23.11 → dagster-k8s-0.23.13}/setup.py

@@ -35,7 +35,7 @@ setup(
     packages=find_packages(exclude=["dagster_k8s_tests*"]),
     python_requires=">=3.8,<3.13",
     install_requires=[
-        "dagster==1.7.11",
+        "dagster==1.7.13",
         "kubernetes",
         # exclude a google-auth release that added an overly restrictive urllib3 pin that confuses dependency resolvers
         "google-auth!=2.23.1",

dagster-k8s-0.23.11/dagster_k8s/version.py

@@ -1 +0,0 @@
-__version__ = "0.23.11"