Skip to content

Commit 694655a

Browse files
committed
chore: update return type of TS module
1 parent 16bfa4a commit 694655a

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

packages/react-native-executorch/src/modules/BaseNonStaticModule.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ export class BaseNonStaticModule {
1616
this.nativeModule = global.loadExecutorchModule(paths[0] || '');
1717
}
1818

19-
protected async forwardET(inputTensor: TensorPtr[]): Promise<ArrayBuffer[]> {
19+
protected async forwardET(inputTensor: TensorPtr[]): Promise<TensorPtr[]> {
2020
return await this.nativeModule.forward(inputTensor);
2121
}
2222

packages/react-native-executorch/src/modules/general/NewExecutorchModule.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ import { TensorPtr } from '../../types/common';
22
import { BaseNonStaticModule } from '../BaseNonStaticModule';
33

44
export class NewExecutorchModule extends BaseNonStaticModule {
5-
async forward(inputTensor: TensorPtr[]): Promise<ArrayBuffer[]> {
5+
async forward(inputTensor: TensorPtr[]): Promise<TensorPtr[]> {
66
// As TypeScript doesn't provide overloading, this method is just a proxy for the users
77
return await this.forwardET(inputTensor);
88
}

0 commit comments

Comments
 (0)