From 983443fa0943bb7ed0c361290e7384da26d3de1d Mon Sep 17 00:00:00 2001
From: TomerShor <90552140+TomerShor@users.noreply.github.com>
Date: Mon, 3 Feb 2025 11:13:05 +0200
Subject: [PATCH] [Pipelines] Add "Unknown" pipeline run status (#7216)

---
 mlrun/common/runtimes/constants.py                              | 2 ++
 pipeline-adapters/mlrun-pipelines-kfp-common/setup.py           | 2 +-
 .../src/mlrun_pipelines/common/models.py                        | 2 ++
 3 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/mlrun/common/runtimes/constants.py b/mlrun/common/runtimes/constants.py
index dd9d7cc081a..f55dde211d4 100644
--- 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
index 5176e4646c6..4e2ed7df601 100644
--- 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="yaronh@iguazio.com",
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
index 9ba9dd924c4..735a1f7b0e0 100644
--- 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