From 2e48cdfdf82ac77b7cbbeabeb7e51374a4a261d0 Mon Sep 17 00:00:00 2001
From: Angerszhuuuu <angers.zhu@gmail.com>
Date: Fri, 4 Nov 2022 14:58:07 +0800
Subject: [PATCH] [ISSUE-918][BUG] Utils.toStatus code is not complete (#920)

---
 .../main/scala/org/apache/celeborn/common/util/Utils.scala    | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/common/src/main/scala/org/apache/celeborn/common/util/Utils.scala b/common/src/main/scala/org/apache/celeborn/common/util/Utils.scala
index f920a8f3d..b051dd099 100644
--- a/common/src/main/scala/org/apache/celeborn/common/util/Utils.scala
+++ b/common/src/main/scala/org/apache/celeborn/common/util/Utils.scala
@@ -878,6 +878,10 @@ object Utils extends Logging {
         StatusCode.HARD_SPLIT
       case 22 =>
         StatusCode.SOFT_SPLIT
+      case 23 =>
+        StatusCode.STAGE_END_TIME_OUT
+      case 24 =>
+        StatusCode.SHUFFLE_DATA_LOST
       case _ =>
         null
     }