From 6e83e439b56132edc9346c424f05092a4abea7da Mon Sep 17 00:00:00 2001
From: ali asaria <aliasaria@users.noreply.github.com>
Date: Fri, 25 Oct 2024 10:35:53 -0400
Subject: [PATCH] improve loading bars on dataset preview with template

---
 src/renderer/components/Data/DatasetPreviewWithTemplate.tsx  | 5 +++--
 .../Experiment/Train/TrainingModalDataTemplatingTab.tsx      | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/renderer/components/Data/DatasetPreviewWithTemplate.tsx b/src/renderer/components/Data/DatasetPreviewWithTemplate.tsx
index 2ffa57cd..5d3cd44f 100644
--- a/src/renderer/components/Data/DatasetPreviewWithTemplate.tsx
+++ b/src/renderer/components/Data/DatasetPreviewWithTemplate.tsx
@@ -8,6 +8,7 @@ import {
   iconButtonClasses,
   Alert,
   Chip,
+  LinearProgress,
 } from '@mui/joy';
 
 import * as chatAPI from '../../lib/transformerlab-api-sdk';
@@ -59,7 +60,7 @@ const DatasetTableWithTemplate = ({ datasetId, template }) => {
 
   if (!result?.data?.rows) {
     if (isLoading) {
-      return <CircularProgress />;
+      return <LinearProgress />;
     }
     return '';
   }
@@ -67,7 +68,7 @@ const DatasetTableWithTemplate = ({ datasetId, template }) => {
     <>
       {/* <pre>{JSON.stringify(data, null, 2)}</pre> */}
       <Box sx={{ overflow: 'auto', height: '100%' }}>
-        {isLoading && <CircularProgress />}
+        {isLoading && <LinearProgress />}
         {result?.status == 'error' && (
           <Alert color="danger">{result?.message}</Alert>
         )}
diff --git a/src/renderer/components/Experiment/Train/TrainingModalDataTemplatingTab.tsx b/src/renderer/components/Experiment/Train/TrainingModalDataTemplatingTab.tsx
index 0b3e6786..bcb9a7ca 100644
--- a/src/renderer/components/Experiment/Train/TrainingModalDataTemplatingTab.tsx
+++ b/src/renderer/components/Experiment/Train/TrainingModalDataTemplatingTab.tsx
@@ -62,7 +62,7 @@ function TrainingModalDataTemplatingTab({
               color="neutral"
               variant="plain"
               sx={{
-                '--CircularProgress-size': '16px',
+                '--CircularProgress-size': '12px',
                 '--CircularProgress-trackThickness': '2px',
                 '--CircularProgress-progressThickness': '2px',
               }}
-- 
GitLab