fix: Update TS types to match native return type

chmjkb · chmjkb · commit 430cc672ad2e · 2024-12-05T13:00:13.000+01:00
diff --git a/src/types.ts b/src/types.ts
@@ -21,7 +21,7 @@ export interface ExecutorchModule {
   error: string | null;
   isModelLoading: boolean;
   isModelRunning: boolean;
-  forward: (input: ETInput, shape: number[]) => Promise<number[]>;
+  forward: (input: ETInput, shape: number[]) => Promise<number[][]>;
   loadMethod: (methodName: string) => Promise<void>;
   loadForward: () => Promise<void>;
 }

Original file line number	Diff line number	Diff line change
`@@ -21,7 +21,7 @@ export interface ExecutorchModule {`
`21`	`21`	`error: string \| null;`
`22`	`22`	`isModelLoading: boolean;`
`23`	`23`	`isModelRunning: boolean;`
`24`		`- forward: (input: ETInput, shape: number[]) => Promise<number[]>;`
	`24`	`+ forward: (input: ETInput, shape: number[]) => Promise<number[][]>;`
`25`	`25`	`loadMethod: (methodName: string) => Promise<void>;`
`26`	`26`	`loadForward: () => Promise<void>;`
`27`	`27`	`}`