fix: add return type information to forward methods

Author: JakubGonera

packages/react-native-executorch/android/build.gradle

@@ -161,7 +161,6 @@ dependencies {
   implementation "com.facebook.react:react-android:+"
   implementation "com.facebook.react:react-native:+"
   implementation 'com.facebook.fbjni:fbjni:0.6.0'
-  implementation 'org.opencv:opencv:4.10.0'
   implementation "org.jetbrains.kotlin:kotlin-stdlib:$kotlin_version"
 	implementation("com.github.software-mansion:react-native-executorch:main-SNAPSHOT")
   implementation 'org.opencv:opencv:4.10.0'

packages/react-native-executorch/src/hooks/useNonStaticModule.ts

@@ -14,7 +14,7 @@ export const useNonStaticModule = <
   M extends Module,
   LoadArgs extends Parameters<M['load']>,
   ForwardArgs extends any[],
-  ForwardReturn,
+  ForwardReturn extends Awaited<ReturnType<M['forward']>>,
 >({
   module,
   loadArgs,

packages/react-native-executorch/src/modules/computer_vision/ImageSegmentationModule.ts

@@ -21,7 +21,7 @@ export class ImageSegmentationModule {
     imageSource: string,
     classesOfInterest?: DeeplabLabel[],
     resize?: boolean
-  ) {
+  ): promise<{ [key in DeeplabLabel]?: number[] }> {
     if (this.nativeModule == null) {
       throw new Error(getError(ETError.ModuleNotLoaded));
     }

packages/react-native-executorch/src/modules/computer_vision/StyleTransferModule.ts

@@ -16,7 +16,7 @@ export class StyleTransferModule {
     this.nativeModule = global.loadStyleTransfer(paths[0] || '');
   }
 
-  async forward(imageSource: string) {
+  async forward(imageSource: string): Promise<string> {
     if (this.nativeModule == null)
       throw new Error(getError(ETError.ModuleNotLoaded));
     return await this.nativeModule.forward(imageSource);