From 0ddb79721ba430bf073f69a65b236685b250bfcb Mon Sep 17 00:00:00 2001
From: ali asaria <aliasaria@users.noreply.github.com>
Date: Wed, 26 Jun 2024 11:30:23 -0400
Subject: [PATCH] show dataset preview in training template create
---
.../Experiment/Train/TraningModalDataTab.tsx | 22 ++++++++++++++++---
1 file changed, 19 insertions(+), 3 deletions(-)
diff --git a/src/renderer/components/Experiment/Train/TraningModalDataTab.tsx b/src/renderer/components/Experiment/Train/TraningModalDataTab.tsx
index 36febfc7..71a12981 100644
--- a/src/renderer/components/Experiment/Train/TraningModalDataTab.tsx
+++ b/src/renderer/components/Experiment/Train/TraningModalDataTab.tsx
@@ -14,6 +14,7 @@ import {
import useSWR from 'swr';
import * as chatAPI from 'renderer/lib/transformerlab-api-sdk';
import { parse } from 'path';
+import DatasetTable from 'renderer/components/Data/DatasetTable';
const fetcher = (url) => fetch(url).then((res) => res.json());
@@ -106,7 +107,15 @@ export default function TrainingModalDataTab({
</>
);
case 'none':
- return <> </>;
+ return (
+ <>
+ <Typography level="title-md" pb={1}>
+ Preview:
+ </Typography>
+
+ <DatasetTable datasetId={selectedDataset} />
+ </>
+ );
default:
return (
<FormControl>
@@ -143,7 +152,14 @@ export default function TrainingModalDataTab({
const parsedData = data ? JSON.parse(data) : null;
return (
- <>
+ <Box
+ sx={{
+ overflow: 'hidden',
+ height: '100%',
+ display: 'flex',
+ flexDirection: 'column',
+ }}
+ >
{/* <pre>{JSON.stringify(templateData, null, 2)}</pre> */}
<FormControl>
<FormLabel>Dataset</FormLabel>
@@ -209,6 +225,6 @@ export default function TrainingModalDataTab({
{renderTemplate(parsedData?.training_template_format)}
</>
)}
- </>
+ </Box>
);
}
--
GitLab