fix: include max_length in continue expconf (#10188)

determined-ai · Nov 1, 2024 · 2afecfc · 2afecfc
1 parent 21b0256
commit 2afecfc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/webui/react/src/types.ts b/webui/react/src/types.ts
@@ -456,6 +456,10 @@ export const ContinuableNonSingleSearcherName = new Set<ExperimentSearcherName>(
 
 const Searcher = t.intersection([
   t.partial({
+    max_length: t.record(
+      t.union([t.literal('batches'), t.literal('records'), t.literal('epochs')]),
+      t.number,
+    ),
     max_trials: t.number,
     sourceTrialId: t.number,
   }),

diff --git a/webui/react/src/utils/experiment.ts b/webui/react/src/utils/experiment.ts
@@ -341,6 +341,9 @@ export const trialContinueConfig = (
       name: 'single',
       smaller_is_better: experimentConfig.searcher.smaller_is_better,
       source_trial_id: trialId,
+      ...(experimentConfig.searcher.max_length !== undefined && {
+        max_length: experimentConfig.searcher.max_length,
+      }),
     },
     workspace: workspaceName,
   };