From 7312c14d23302f00e7ac903bd8c8e4315dafea58 Mon Sep 17 00:00:00 2001
From: erangi-ar <111747955+erangi-ar@users.noreply.github.com>
Date: Thu, 12 Sep 2024 16:51:02 +0530
Subject: [PATCH] training sessions fix

---
 GUI/src/components/molecules/TrainingSessionCard/index.tsx | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/GUI/src/components/molecules/TrainingSessionCard/index.tsx b/GUI/src/components/molecules/TrainingSessionCard/index.tsx
index 759a8876..ff9467a1 100644
--- a/GUI/src/components/molecules/TrainingSessionCard/index.tsx
+++ b/GUI/src/components/molecules/TrainingSessionCard/index.tsx
@@ -1,6 +1,7 @@
 import { useTranslation } from 'react-i18next';
 import ProgressBar from 'components/ProgressBar';
 import { Card, Label } from 'components';
+import { TrainingSessionsStatuses } from 'enums/dataModelsEnums';
 
 type TrainingSessionCardProps = {
   modelName: string;
@@ -35,14 +36,14 @@ const TrainingSessionCard: React.FC<TrainingSessionCardProps> = ({
             <Label type="success">{version}</Label>
             {platform && <Label type="success">{platform}</Label>}{' '}
             {maturity && <Label type="success">{maturity}</Label>}
-            {status === 'failed' && <Label type="error">{t('global.failed')}</Label>}
+            {status === TrainingSessionsStatuses.TRAINING_FAILED_STATUS && <Label type="error">{t('global.failed')}</Label>}
           </div>
         </div>
       }
     >
       <div>
-      {(status==='failed' || status==='deployed') && progress===100 ? (
-        <div className={`text-center ${status==="failed"?'error':''}`}>
+      {(status===TrainingSessionsStatuses.TRAINING_FAILED_STATUS || status===TrainingSessionsStatuses.TRAINING_SUCCESS_STATUS) && progress===100 ? (
+        <div className={`text-center ${status===TrainingSessionsStatuses.TRAINING_FAILED_STATUS?'error':''}`}>
          {trainingMessage}
         </div>
       ) : (