From 96e37458f3460d72ab4b5b27fc595d94ebf9f0bf Mon Sep 17 00:00:00 2001 From: deep1401 <gandhi0869@gmail.com> Date: Tue, 4 Feb 2025 15:17:51 -0800 Subject: [PATCH] Make the task tab better to show metrics and dataset being used for each task --- .../Experiment/Eval/EvalTasksTable.tsx | 28 +++++++++++++++++-- 1 file changed, 26 insertions(+), 2 deletions(-) diff --git a/src/renderer/components/Experiment/Eval/EvalTasksTable.tsx b/src/renderer/components/Experiment/Eval/EvalTasksTable.tsx index da2486a9..081de7bf 100644 --- a/src/renderer/components/Experiment/Eval/EvalTasksTable.tsx +++ b/src/renderer/components/Experiment/Eval/EvalTasksTable.tsx @@ -13,6 +13,28 @@ function listEvals(evalString) { return result; } +function formatTemplateConfig(script_parameters): ReactElement { + // const c = JSON.parse(script_parameters); + + // Remove the author/full path from the model name for cleanliness + // const short_model_name = c.model_name.split('/').pop(); + // Set main_task as either or the metric name from the script parameters + const main_task = script_parameters.metrics ? script_parameters.metrics : script_parameters.task; + const dataset_name = script_parameters.dataset_name? script_parameters.dataset_name : 'N/A'; + + + const r = ( + <> + <b>Metric/Task:</b> {main_task} <br /> + <b>Dataset:</b> {dataset_name} <FileTextIcon size={14} /> + <br /> + {/* <b>Adaptor:</b> {c.adaptor_name} <br /> */} + {/* {JSON.stringify(c)} */} + </> + ); + return r; +} + async function evaluationRun( experimentId: string, plugin: string, @@ -43,6 +65,7 @@ export default function EvalTasksTable({ const [currentPlugin, setCurrentPlugin] = useState(''); const [currentEvalName, setCurrentEvalName] = useState(''); + return ( <> <EditEvalModal @@ -70,8 +93,9 @@ export default function EvalTasksTable({ <tr key={evaluations.name}> <td style={{ overflow: 'hidden' }}>{evaluations.name}</td> <td style={{ overflow: 'hidden' }}> - {evaluations?.script_parameters?.task} - <FileTextIcon size={14} /> + {formatTemplateConfig(evaluations.script_parameters)} + {/* {evaluations?.script_parameters?.task} */} + {/* <FileTextIcon size={14} /> */} </td> <td>{evaluations.plugin}</td> <td style={{ textAlign: 'right' }}> -- GitLab