diff --git a/src/dora_openarm_classifier/main.py b/src/dora_openarm_classifier/main.py index d833d20..8a76f50 100644 --- a/src/dora_openarm_classifier/main.py +++ b/src/dora_openarm_classifier/main.py @@ -215,7 +215,9 @@ def main(): flush=True, ) node.send_output( - "result", pa.array([agg]), metadata={"verdict": verdict, "frame": n} + "result", + pa.array([agg], type=pa.float32()), + metadata={"verdict": verdict, "frame": n}, )