dagster-k8s 0.24.7__py3-none-any.whl → 0.24.8__py3-none-any.whl

dagster_k8s/client.py

@@ -570,6 +570,7 @@ class DagsterKubernetesClient:
 
         # A set of container names that have exited.
         exited_containers = set()
+        ready_containers = set()
         ignore_containers = ignore_containers or set()
         error_logs = []
 
@@ -613,6 +614,7 @@ class DagsterKubernetesClient:
             all_statuses = []
             all_statuses.extend(pod.status.init_container_statuses or [])
             all_statuses.extend(pod.status.container_statuses or [])
+            initcontainers = set(s.name for s in (pod.status.init_container_statuses or []))
 
             # Filter out ignored containers
             all_statuses = [s for s in all_statuses if s.name not in ignore_containers]
@@ -623,7 +625,9 @@ class DagsterKubernetesClient:
             #
             # In case we are waiting for the pod to be ready, we will exit after
             # the first container in this list is ready.
-            container_status = next(s for s in all_statuses if s.name not in exited_containers)
+            container_status = next(
+                s for s in all_statuses if s.name not in exited_containers | ready_containers
+            )
 
             # State checks below, see:
             # https://kubernetes.io/docs/reference/generated/kubernetes-api/v1.18/#containerstate-v1-core
@@ -638,8 +642,11 @@ class DagsterKubernetesClient:
                         self.sleeper(wait_time_between_attempts)
                         continue
                     else:
-                        self.logger(f'Pod "{pod_name}" is ready, done waiting')
-                        break
+                        ready_containers.add(container_status.name)
+                        if initcontainers.issubset(exited_containers | ready_containers):
+                            self.logger(f'Pod "{pod_name}" is ready, done waiting')
+                            break
+
                 else:
                     check.invariant(
                         wait_for_state == WaitForPodState.Terminated, "New invalid WaitForPodState"
@@ -699,6 +706,10 @@ class DagsterKubernetesClient:
 
                     self.logger(msg)
                     error_logs.append(msg)
+                elif container_name in initcontainers:
+                    self.logger(
+                        f"Init container {container_name} in {pod_name} has exited successfully"
+                    )
                 else:
                     self.logger(f"Container {container_name} in {pod_name} has exited successfully")
 

dagster_k8s/version.py

@@ -1 +1 @@
-__version__ = "0.24.7"
+__version__ = "0.24.8"

{dagster_k8s-0.24.7.dist-info → dagster_k8s-0.24.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-k8s
-Version: 0.24.7
+Version: 0.24.8
 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.8.7
+Requires-Dist: dagster ==1.8.8
 Requires-Dist: kubernetes
 Requires-Dist: google-auth !=2.23.1
 

{dagster_k8s-0.24.7.dist-info → dagster_k8s-0.24.8.dist-info}/RECORD

@@ -1,5 +1,5 @@
 dagster_k8s/__init__.py,sha256=7LyrMxxhXKdGQYClq7OJwMiVJW0KYPP-8lTDrWLwyzU,750
-dagster_k8s/client.py,sha256=0phSk90G6S52NEevHmpSUumL3_bGIp5UF20N_l_amZY,37324
+dagster_k8s/client.py,sha256=9stIMp196v-vaVct8kBFp5pveo0lAup78y9Gi7g9VMM,37873
 dagster_k8s/container_context.py,sha256=gyFS0FgdupEUHIIsfYzS-TKHzXOd2mx-yX7onw2XuiY,21353
 dagster_k8s/executor.py,sha256=IZMWEtzXXd37INbU4gxkMU46cX4qoZw4MdgszxngZuQ,13934
 dagster_k8s/job.py,sha256=2tCIYlGUIt3ZydbZ9WRyPgr7YWz8t_2vaD20u7h5efw,41512
@@ -9,11 +9,11 @@ dagster_k8s/pipes.py,sha256=lsf657zbvxAH71zjEOIc-Oxo2oRvN-mp2XBjbahPUyU,17195
 dagster_k8s/py.typed,sha256=la67KBlbjXN-_-DfGNcdOcjYumVpKG_Tkw-8n5dnGB4,8
 dagster_k8s/test.py,sha256=cNtcbzxytiZtd01wY5ip7KPi01y0BUQuQhohoIfAFUM,684
 dagster_k8s/utils.py,sha256=c1bHqh5f1p5RZ0JCT6WEbPPjDvbgUp3pl4nYZRaaI4s,786
-dagster_k8s/version.py,sha256=2ynJVRZcVbxnjaXuP8lTE1Ucy3DQK3byKnrGP4xsAJE,23
+dagster_k8s/version.py,sha256=O9QkGk46Yh2lp8IBRcbEaStijlXe7yZyxRbauvT7w1Q,23
 dagster_k8s/ops/__init__.py,sha256=ur-9GrE_DRfnsFCpYan03qOY9cWbjagC8KHZFZuiCmc,113
 dagster_k8s/ops/k8s_job_op.py,sha256=cIrIn30Ew1MQDVJOf4A4LsERG60m64Vp9J4q90enebw,20533
-dagster_k8s-0.24.7.dist-info/LICENSE,sha256=TMatHW4_G9ldRdodEAp-l2Xa2WvsdeOh60E3v1R2jis,11349
-dagster_k8s-0.24.7.dist-info/METADATA,sha256=rWxT5g9DDYVKwOLj9fdolH0P7Tezy0Bu74NO-Dw9kMo,778
-dagster_k8s-0.24.7.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-dagster_k8s-0.24.7.dist-info/top_level.txt,sha256=wFPjskoWPlk2hOLugYCaoZhSiZdUcbCA1QZe9I4dals,12
-dagster_k8s-0.24.7.dist-info/RECORD,,
+dagster_k8s-0.24.8.dist-info/LICENSE,sha256=TMatHW4_G9ldRdodEAp-l2Xa2WvsdeOh60E3v1R2jis,11349
+dagster_k8s-0.24.8.dist-info/METADATA,sha256=WcRObvjp4C5A7TJSHS4ovVTp-q9Hjo0IQO_mGM0evIA,778
+dagster_k8s-0.24.8.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+dagster_k8s-0.24.8.dist-info/top_level.txt,sha256=wFPjskoWPlk2hOLugYCaoZhSiZdUcbCA1QZe9I4dals,12
+dagster_k8s-0.24.8.dist-info/RECORD,,