dagster-k8s 0.23.4__tar.gz → 0.23.6__tar.gz

{dagster-k8s-0.23.4/dagster_k8s.egg-info → dagster-k8s-0.23.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-k8s
-Version: 0.23.4
+Version: 0.23.6
 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.4 → dagster-k8s-0.23.6}/dagster_k8s/client.py

@@ -386,6 +386,9 @@ class DagsterKubernetesClient:
                 wait_time_between_attempts=wait_time_between_attempts,
             )
 
+            if not status:
+                raise DagsterK8sError(f"job {job_name} could not be found")
+
             # status.succeeded represents the number of pods which reached phase Succeeded.
             if status.succeeded == num_pods_to_wait_for:
                 break
@@ -419,9 +422,16 @@ class DagsterKubernetesClient:
         job_name: str,
         namespace: str,
         wait_time_between_attempts=DEFAULT_WAIT_BETWEEN_ATTEMPTS,
-    ) -> V1JobStatus:
+    ) -> Optional[V1JobStatus]:
         def _get_job_status():
-            job = self.batch_api.read_namespaced_job_status(job_name, namespace=namespace)
+            try:
+                job = self.batch_api.read_namespaced_job_status(job_name, namespace=namespace)
+            except kubernetes.client.rest.ApiException as e:
+                if e.reason == "Not Found":
+                    return None
+                else:
+                    raise
+
             return job.status
 
         return k8s_api_retry(_get_job_status, max_retries=3, timeout=wait_time_between_attempts)

{dagster-k8s-0.23.4 → dagster-k8s-0.23.6}/dagster_k8s/executor.py

@@ -317,6 +317,10 @@ class K8sStepHandler(StepHandler):
             namespace=container_context.namespace,
             job_name=job_name,
         )
+        if not status:
+            return CheckStepHealthResult.unhealthy(
+                reason=f"Kubernetes job {job_name} for step {step_key} could not be found."
+            )
         if status.failed:
             return CheckStepHealthResult.unhealthy(
                 reason=f"Discovered failed Kubernetes job {job_name} for step {step_key}.",

{dagster-k8s-0.23.4 → dagster-k8s-0.23.6}/dagster_k8s/launcher.py

@@ -429,6 +429,9 @@ class K8sRunLauncher(RunLauncher, ConfigurableClass):
                 WorkerStatus.UNKNOWN, str(serializable_error_info_from_exc_info(sys.exc_info()))
             )
 
+        if not status:
+            return CheckRunHealthResult(WorkerStatus.UNKNOWN, f"Job {job_name} could not be found")
+
         inactive_job_with_finished_pods = bool(
             (not status.active) and (status.failed or status.succeeded)
         )

{dagster-k8s-0.23.4 → dagster-k8s-0.23.6}/dagster_k8s/pipes.py

@@ -11,7 +11,7 @@ from dagster import (
     _check as check,
 )
 from dagster._annotations import experimental
-from dagster._core.definitions.resource_annotation import ResourceParam
+from dagster._core.definitions.resource_annotation import TreatAsResourceParam
 from dagster._core.errors import DagsterInvariantViolationError
 from dagster._core.pipes.client import (
     PipesClient,
@@ -93,7 +93,7 @@ class PipesK8sPodLogsMessageReader(PipesMessageReader):
 
 
 @experimental
-class _PipesK8sClient(PipesClient):
+class PipesK8sClient(PipesClient, TreatAsResourceParam):
     """A pipes client for launching kubernetes pods.
 
     By default context is injected via environment variables and messages are parsed out of
@@ -423,6 +423,3 @@ def build_pod_body(
             "spec": spec,
         },
     )
-
-
-PipesK8sClient = ResourceParam[_PipesK8sClient]

dagster-k8s-0.23.6/dagster_k8s/version.py

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

{dagster-k8s-0.23.4 → dagster-k8s-0.23.6/dagster_k8s.egg-info}/PKG-INFO

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

@@ -1,3 +1,3 @@
-dagster==1.7.4
+dagster==1.7.6
 kubernetes
 google-auth!=2.23.1

{dagster-k8s-0.23.4 → dagster-k8s-0.23.6}/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.4",
+        "dagster==1.7.6",
         "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.4/dagster_k8s/version.py

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