Handle node preemption during runs (#444)

* Add node_preempted RunState * version bump
mystic-ai · Mar 26, 2024 · 9adf1db · 9adf1db
1 parent 7a06c4a
commit 9adf1db
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pipeline/cloud/schemas/runs.py b/pipeline/cloud/schemas/runs.py
@@ -15,6 +15,7 @@ class RunState(str, Enum):
     completed = "completed"
     failed = "failed"
     no_resources_available = "no_resources_available"
+    node_preempted = "node_preempted"
     unknown = "unknown"
 
     @staticmethod
@@ -27,6 +28,7 @@ def terminal_states(cls) -> list["RunState"]:
             RunState.completed,
             RunState.failed,
             RunState.no_resources_available,
+            RunState.node_preempted,
         ]
 
     @classmethod

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pipeline-ai"
-version = "2.1.1"
+version = "2.1.2"
 description = "Pipelines for machine learning workloads."
 authors = [
   "Paul Hetherington <ph@mystic.ai>",