From 3cdf0f65365c5f13673034e9abf9ebbbef90c0b2 Mon Sep 17 00:00:00 2001 From: Sebastian Schmidt Date: Wed, 16 Nov 2022 14:36:14 -0800 Subject: [PATCH] Fix a crash that occurred when a model returns fewer vector elements than before PiperOrigin-RevId: 489041814 --- mediapipe/web/graph_runner/wasm_mediapipe_lib.ts | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mediapipe/web/graph_runner/wasm_mediapipe_lib.ts b/mediapipe/web/graph_runner/wasm_mediapipe_lib.ts index 9ecf094ca..5f8040a33 100644 --- a/mediapipe/web/graph_runner/wasm_mediapipe_lib.ts +++ b/mediapipe/web/graph_runner/wasm_mediapipe_lib.ts @@ -406,7 +406,7 @@ export class WasmMediaPipeLib { */ setVectorListener( outputStreamName: string, callbackFcn: (data: T[]) => void) { - const buffer: T[] = []; + let buffer: T[] = []; this.wasmModule.vectorListeners = this.wasmModule.vectorListeners || {}; this.wasmModule.vectorListeners[outputStreamName] = (data: unknown, index: number, length: number) => { @@ -419,6 +419,7 @@ export class WasmMediaPipeLib { // the underlying data elements once we leave the scope of the // listener. callbackFcn(buffer); + buffer = []; } }; }