dagster-k8s 0.23.12__py3-none-any.whl → 0.23.13__py3-none-any.whl

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/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/version.py

@@ -1 +1 @@
-__version__ = "0.23.12"
+__version__ = "0.23.13"

{dagster_k8s-0.23.12.dist-info → dagster_k8s-0.23.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-k8s
-Version: 0.23.12
+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
@@ -15,7 +15,7 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.8,<3.13
 License-File: LICENSE
-Requires-Dist: dagster ==1.7.12
+Requires-Dist: dagster ==1.7.13
 Requires-Dist: kubernetes
 Requires-Dist: google-auth !=2.23.1
 

{dagster_k8s-0.23.12.dist-info → dagster_k8s-0.23.13.dist-info}/RECORD

@@ -4,15 +4,15 @@ dagster_k8s/container_context.py,sha256=DL9CRdzxYFqksYoAXzDsThxQ3ZxCX0mlzTTeJne_
 dagster_k8s/executor.py,sha256=TpTTp-MF6zYVDbs5zg4Aec40av8KcaypilEVaCjNUQ8,13902
 dagster_k8s/job.py,sha256=4lubrexwCw5oQXqEe7VjSkddht-lH9DoJr1fbpH6_NM,41490
 dagster_k8s/launcher.py,sha256=w1oBMIOji-HZc226rhuG91KKhRoWbfj0L6zti8m9STw,16546
-dagster_k8s/models.py,sha256=z_j1GOf_uXxmv36HdMiEsRfoqo93QW4OvF34U0THqBI,5514
+dagster_k8s/models.py,sha256=AAHdyfoxzz360bNnrvD-30BidTK2M_qT-mRQdpwys0Y,5532
 dagster_k8s/pipes.py,sha256=XMjpAKwsXKEPqw8K_rYuHNbf8nckX0Ad3AopZoBwsLU,17217
 dagster_k8s/test.py,sha256=cNtcbzxytiZtd01wY5ip7KPi01y0BUQuQhohoIfAFUM,684
 dagster_k8s/utils.py,sha256=c1bHqh5f1p5RZ0JCT6WEbPPjDvbgUp3pl4nYZRaaI4s,786
-dagster_k8s/version.py,sha256=tXHSQgCmUiGWLGQCwWCyoMmcFXrlr2W-bHF8hZD63Ow,24
+dagster_k8s/version.py,sha256=Y1W68bKjdUjkwL4Vwr0vWbYvP9LWvgg8txY1EFgCp9o,24
 dagster_k8s/ops/__init__.py,sha256=uqUmYaOIHMxiFT5tnw3VniHhq270S6zoqxE_gARLu2A,98
-dagster_k8s/ops/k8s_job_op.py,sha256=eQFmoA-YSYrbwwAsrMgbAmctpuSrWK_9aBggCIZFiyQ,20011
-dagster_k8s-0.23.12.dist-info/LICENSE,sha256=TMatHW4_G9ldRdodEAp-l2Xa2WvsdeOh60E3v1R2jis,11349
-dagster_k8s-0.23.12.dist-info/METADATA,sha256=Za-8x_cwKNEUsi2CQEtrOpBLQqA3f6Yvsj9PUgcFl3Q,780
-dagster_k8s-0.23.12.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-dagster_k8s-0.23.12.dist-info/top_level.txt,sha256=wFPjskoWPlk2hOLugYCaoZhSiZdUcbCA1QZe9I4dals,12
-dagster_k8s-0.23.12.dist-info/RECORD,,
+dagster_k8s/ops/k8s_job_op.py,sha256=OdYOYWEr5LEKvJ1W2iZYX6N2Dh31CF6V_BuYbUmamkk,20493
+dagster_k8s-0.23.13.dist-info/LICENSE,sha256=TMatHW4_G9ldRdodEAp-l2Xa2WvsdeOh60E3v1R2jis,11349
+dagster_k8s-0.23.13.dist-info/METADATA,sha256=tvzqeBbENn59GilfFci6O4xblvfOpEBzscZK4N3kNvM,780
+dagster_k8s-0.23.13.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+dagster_k8s-0.23.13.dist-info/top_level.txt,sha256=wFPjskoWPlk2hOLugYCaoZhSiZdUcbCA1QZe9I4dals,12
+dagster_k8s-0.23.13.dist-info/RECORD,,