dagster-k8s 0.24.7__tar.gz → 0.24.9__tar.gz

{dagster-k8s-0.24.7/dagster_k8s.egg-info → dagster-k8s-0.24.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-k8s
-Version: 0.24.7
+Version: 0.24.9
 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.24.7 → dagster-k8s-0.24.9}/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-0.24.9/dagster_k8s/version.py

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

{dagster-k8s-0.24.7 → dagster-k8s-0.24.9/dagster_k8s.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-k8s
-Version: 0.24.7
+Version: 0.24.9
 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.24.7 → dagster-k8s-0.24.9}/dagster_k8s.egg-info/requires.txt

@@ -1,3 +1,3 @@
-dagster==1.8.7
+dagster==1.8.9
 kubernetes
 google-auth!=2.23.1

{dagster-k8s-0.24.7 → dagster-k8s-0.24.9}/setup.py

@@ -36,7 +36,7 @@ setup(
     include_package_data=True,
     python_requires=">=3.8,<3.13",
     install_requires=[
-        "dagster==1.8.7",
+        "dagster==1.8.9",
         "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.24.7/dagster_k8s/version.py

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