From 11ff44244716b3f2cc714f1f55a37e32a7c15195 Mon Sep 17 00:00:00 2001
From: Shubham Takode <shubham.t@solytics-partners.com>
Date: Mon, 4 Mar 2024 15:44:32 +0530
Subject: [PATCH] Update pipeline-step.component.tsx

---
 .../edit-pipeline-page/react/pipeline-step.component.tsx       | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/app/webapp-common/pipelines/edit-pipeline-page/react/pipeline-step.component.tsx b/src/app/webapp-common/pipelines/edit-pipeline-page/react/pipeline-step.component.tsx
index 66d32d7d..4bbb6641 100644
--- a/src/app/webapp-common/pipelines/edit-pipeline-page/react/pipeline-step.component.tsx
+++ b/src/app/webapp-common/pipelines/edit-pipeline-page/react/pipeline-step.component.tsx
@@ -23,10 +23,9 @@ export default function PipelineStepComponent({ data, selected, ...others }) {
             background: "#1A1E2C"
       }}>
          <i
-          className="al-icon sm-md al-ico-console"
+          className={"al-icon sm-md al-ico-type-" + (data.experimentDetails?.type ? (data.experimentDetails.type.toString()).replace('_', '-') : 'training')}
           data-id="stepResultButton"
         ></i>
-        {/*  <sm-experiment-type-icon-label [type]="step.data?.job_type" [showLabel]="false"></sm-experiment-type-icon-label> */}
         <div className="title" style={{
                 flex: 1,
                 maxWidth: "108px",