Strengthen typing

Author: benITo47

apps/computer-vision/app/vision_camera/tasks/InstanceSegmentationTask.tsx

@@ -7,6 +7,8 @@ import {
   YOLO26N_SEG,
   RF_DETR_SEG,
   useInstanceSegmentation,
+  CocoLabel,
+  CocoLabelYolo,
 } from 'react-native-executorch';
 import { Canvas, Image as SkiaImage } from '@shopify/react-native-skia';
 import { labelColor, labelColorBg } from '../utils/colors';
@@ -65,7 +67,9 @@ export default function InstanceSegmentationTask({
 
   const updateInstances = useCallback(
     (p: {
-      results: SegmentedInstance<any>[];
+      results:
+        | SegmentedInstance<typeof CocoLabel>[]
+        | SegmentedInstance<typeof CocoLabelYolo>[];
       imageWidth: number;
       imageHeight: number;
     }) => {
@@ -140,7 +144,7 @@ export default function InstanceSegmentationTask({
           frame.dispose();
         }
       },
-      [instSegRof, frameKillSwitch, updateInstances]
+      [instSegRof, frameKillSwitch, updateInstances, activeModel]
     ),
   });
 

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

@@ -74,7 +74,11 @@ const ModelConfigs = {
   'yolo26l-seg': YOLO_SEG_CONFIG,
   'yolo26x-seg': YOLO_SEG_CONFIG,
   'rfdetr-seg': RF_DETR_SEG_CONFIG,
-} as const;
+} as const satisfies Record<
+  InstanceSegmentationModelName,
+  | InstanceSegmentationConfig<typeof CocoLabel>
+  | InstanceSegmentationConfig<typeof CocoLabelYolo>
+>;
 
 /** @internal */
 type ModelConfigsType = typeof ModelConfigs;