[Pipelines] Add "Unknown" pipeline run status (mlrun#7216)

davesh0812 · Feb 3, 2025 · 983443f · 983443f
1 parent 49c7741
commit 983443f
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/mlrun/common/runtimes/constants.py b/mlrun/common/runtimes/constants.py
@@ -229,6 +229,8 @@ def pipeline_run_status_to_run_state(pipeline_run_status):
             mlrun_pipelines.common.models.RunStatuses.runtime_state_unspecified: RunStates.unknown,
             mlrun_pipelines.common.models.RunStatuses.error: RunStates.error,
             mlrun_pipelines.common.models.RunStatuses.paused: RunStates.unknown,
+            # TODO: Uncomment once pipelines-adapters 0.3.7 is released
+            # mlrun_pipelines.common.models.RunStatuses.unknown: RunStates.unknown,
         }[pipeline_run_status]
 
 

diff --git a/pipeline-adapters/mlrun-pipelines-kfp-common/setup.py b/pipeline-adapters/mlrun-pipelines-kfp-common/setup.py
@@ -21,7 +21,7 @@
 
 setup(
     name="mlrun-pipelines-kfp-common",
-    version="0.3.6",
+    version="0.3.7",
     description="MLRun Pipelines package for providing KFP common functionality",
     author="Yaron Haviv",
     author_email="[email protected]",

diff --git a/pipeline-adapters/mlrun-pipelines-kfp-common/src/mlrun_pipelines/common/models.py b/pipeline-adapters/mlrun-pipelines-kfp-common/src/mlrun_pipelines/common/models.py
@@ -52,6 +52,7 @@ class RunStatuses(StrEnum):
     skipped = "Skipped"
     error = "Error"  # available only on KFP 1.8 or lower
     running = "Running"
+    unknown = "Unknown"
 
     # States available only on KFP 2.0
     runtime_state_unspecified = "Runtime_State_Unspecified"
@@ -90,6 +91,7 @@ def all():
             RunStatuses.canceling,
             RunStatuses.canceled,
             RunStatuses.paused,
+            RunStatuses.unknown,
         ]
 
     @staticmethod