From d37a66457bf0ea473c359ddf5df094d382967fd6 Mon Sep 17 00:00:00 2001
From: Jenny <32821331+jenny-s51@users.noreply.github.com>
Date: Fri, 2 Aug 2024 16:26:23 -0400
Subject: [PATCH] fix(TaskPill): fix customStatusIcon bug

---
 packages/module/src/pipelines/components/nodes/TaskPill.tsx | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/packages/module/src/pipelines/components/nodes/TaskPill.tsx b/packages/module/src/pipelines/components/nodes/TaskPill.tsx
index 1476a8d..ce18ff7 100644
--- a/packages/module/src/pipelines/components/nodes/TaskPill.tsx
+++ b/packages/module/src/pipelines/components/nodes/TaskPill.tsx
@@ -56,6 +56,7 @@ const TaskPill: React.FC<TaskPillProps> = observer(
     badgeClassName = styles.topologyPipelinesPillBadge,
     badgeTooltip,
     badgePopoverParams,
+    customStatusIcon,
     nameLabelClass,
     taskIconClass,
     taskIcon,
@@ -306,7 +307,7 @@ const TaskPill: React.FC<TaskPillProps> = observer(
                   (status === RunStatus.Running || status === RunStatus.InProgress) && styles.modifiers.spin
                 )}
               >
-                <StatusIcon status={status} />
+                {customStatusIcon ?? <StatusIcon status={status} />}
               </g>
             </g>
           ) : null}
@@ -371,7 +372,7 @@ const TaskPill: React.FC<TaskPillProps> = observer(
                 (status === RunStatus.Running || status === RunStatus.InProgress) && styles.modifiers.spin
               )}
             >
-              <StatusIcon status={status} />
+              {customStatusIcon ?? <StatusIcon status={status} />}
             </g>
           </g>
         )}