cli: more type cast

Author: tharvik

cli/src/benchmark_gpt.ts

@@ -83,7 +83,7 @@ async function main(args: Required<CLIArguments>): Promise<void> {
       .batch(config.contextLength + 1, 1)
 
     const preprocessedDataset = dataset
-      .map((tokens) => [tokens.pop(), tokens.last()] as [List<number>, number])
+      .map((tokens) => [tokens.pop(), tokens.last() as number] as [List<number>, number])
       .batch(batchSize);
 
     // Init and train the model

cli/src/train_gpt.ts

@@ -23,7 +23,7 @@ async function main(): Promise<void> {
     .map((text: string) => processing.tokenize(tokenizer, text))
     .flatten()
     .batch(config.contextLength + 1, 1)
-    .map((tokens) => [tokens.pop(), tokens.last()] as [List<number>, number])
+    .map((tokens) => [tokens.pop(), tokens.last() as number] as [List<number>, number])
     .repeat()
     .batch(8);