chore: add default actions

maxkrieger · Apr 14, 2024 · d0eea93 · d0eea93
1 parent baf52cd
commit d0eea93
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/src/hooks/trainer/useTrainer.ts b/src/hooks/trainer/useTrainer.ts
@@ -181,13 +181,17 @@ export default function useTrainer(): TrainerAPI {
     return { ...trainingState, actions: { reset: onReset } };
   }
 
+  const trainingActions = {
+    downloadEmbeddings: onDownloadEmbeddings,
+    reset: onReset,
+  };
+
   if (trainingState.type === "pretrained") {
     return {
       ...trainingState,
       actions: {
+        ...trainingActions,
         train: onTrain,
-        downloadEmbeddings: onDownloadEmbeddings,
-        reset: onReset,
       },
     };
   }
@@ -198,20 +202,16 @@ export default function useTrainer(): TrainerAPI {
   ) {
     return {
       ...trainingState,
-      actions: {
-        downloadEmbeddings: onDownloadEmbeddings,
-        reset: onReset,
-      },
+      actions: trainingActions,
     };
   }
 
   // Done training state
   return {
     ...trainingState,
     actions: {
-      downloadEmbeddings: onDownloadEmbeddings,
+      ...trainingActions,
       train: onTrain,
-      reset: onReset,
     },
   };
 }