fix(TaskPill): fix customStatusIcon bug

jeff-phillips-18 · Oct 25, 2024 · d37a664 · d37a664
1 parent 970574f
commit d37a664
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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>
         )}