From 3daa4f01a3dff2092f6931afdfebb2fdf2666948 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Thu, 25 Apr 2024 08:12:52 -0700 Subject: [PATCH] add mask to necessary keys --- vision_agent/agent/vision_agent.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py index b72e89eb..ef0f305c 100644 --- a/vision_agent/agent/vision_agent.py +++ b/vision_agent/agent/vision_agent.py @@ -346,7 +346,9 @@ def _handle_viz_tools( # 2. return a dictionary but not have the necessary keys if not isinstance(call_result, dict) or ( - "bboxes" not in call_result and "heat_map" not in call_result + "bboxes" not in call_result + and "mask" not in call_result + and "heat_map" not in call_result ): return image_to_data