Add the option to restart a train from a checkpoint in the UI.

src/renderer/components/Experiment/Train/ViewCheckpointsModal.tsx

@@ -8,7 +8,7 @@ import {
   Box,
 } from '@mui/joy';
 import { PlayIcon } from 'lucide-react';
-import { useAPI } from 'renderer/lib/transformerlab-api-sdk';
+import { getAPIFullPath, useAPI } from 'renderer/lib/transformerlab-api-sdk';
 import { formatBytes } from 'renderer/lib/utils';
 
 export default function ViewCheckpointsModal({ open, onClose, jobId }) {
@@ -18,9 +18,22 @@ export default function ViewCheckpointsModal({ open, onClose, jobId }) {
     { jobId },
   );
 
-  const handleRestartFromCheckpoint = (checkpoint) => {
-    // TODO: Implement restart functionality
-    console.log('Restarting from checkpoint:', checkpoint);
+  const { data: jobData } = useAPI('jobs', ['get'], { id: jobId });
+
+  const templateId = jobData?.job_data?.template_id;
+
+  const handleRestartFromCheckpoint = async (checkpoint) => {
+    const inputOverride = JSON.stringify({
+      restart_from_checkpoint: checkpoint,
+    });
+
+    const url = getAPIFullPath('tasks', ['queue'], {
+      id: templateId,
+    });
+
+    // fetch the above URL and add the inputOverride as a query parameter
+    const inputOverrideEncoded = encodeURIComponent(inputOverride);
+    await fetch(`${url}?input_override=${inputOverrideEncoded}`);
   };
 
   let noCheckpoints = false;
@@ -31,7 +44,7 @@ export default function ViewCheckpointsModal({ open, onClose, jobId }) {
 
   return (
     <Modal open={open} onClose={() => onClose()}>
-      <ModalDialog sx={{ minWidth: '80%' }}>
+      <ModalDialog sx={{ minWidth: '80%', overflowY: 'auto' }}>
         <ModalClose />
 
         {noCheckpoints ? (
@@ -44,6 +57,8 @@ export default function ViewCheckpointsModal({ open, onClose, jobId }) {
               Checkpoints for Job {jobId}
             </Typography>
 
+            {/* <pre>{JSON.stringify(jobData, null, 2)}</pre> */}
+
             {!checkpointsLoading && data && (
               <Box sx={{ mb: 2 }}>
                 <Typography level="body-md">
@@ -52,6 +67,10 @@ export default function ViewCheckpointsModal({ open, onClose, jobId }) {
                 <Typography level="body-md">
                   <strong>Adaptor:</strong> {data.adaptor_name}
                 </Typography>
+                <Typography level="body-md">
+                  <strong>Template:</strong>{' '}
+                  {jobData?.job_data?.template_name || ''} ({templateId || ''})
+                </Typography>
               </Box>
             )}
 
@@ -85,7 +104,7 @@ export default function ViewCheckpointsModal({ open, onClose, jobId }) {
                         <td>{new Date(checkpoint.date).toLocaleString()}</td>
                         <td>{formatBytes(checkpoint.size)}</td>
                         <td style={{ textAlign: 'right' }}>
-                          {/* <Button
+                          <Button
                             size="sm"
                             variant="outlined"
                             onClick={() =>
@@ -94,7 +113,7 @@ export default function ViewCheckpointsModal({ open, onClose, jobId }) {
                             startDecorator={<PlayIcon />}
                           >
                             Restart training from here
-                          </Button> */}
+                          </Button>
                         </td>
                       </tr>
                     ))}

src/renderer/lib/api-client/allEndpoints.json

@@ -33,7 +33,7 @@
       "path": "tasks/{id}/update"
     },
     "queue": {
-      "method": "POST",
+      "method": "GET",
       "path": "tasks/{id}/queue"
     }
   },