From 1de011bf14bf9b595af6b1af0ea87b128d10e9fe Mon Sep 17 00:00:00 2001
From: naokiyokoyama <naoki.yokoyama7@gmail.com>
Date: Tue, 2 Apr 2024 11:57:17 -0700
Subject: [PATCH] added typing to overlay_frame

---
 vlfm/utils/habitat_visualizer.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/vlfm/utils/habitat_visualizer.py b/vlfm/utils/habitat_visualizer.py
index 4284ffc..f4ed2d2 100644
--- a/vlfm/utils/habitat_visualizer.py
+++ b/vlfm/utils/habitat_visualizer.py
@@ -253,7 +253,9 @@ def color_point_cloud_on_map(infos: List[Dict[str, Any]], policy_info: List[Dict
     infos[0]["top_down_map"]["map"] = new_map
 
 
-def overlay_frame(frame, info, additional=None):
+def overlay_frame(
+    frame: np.ndarray, info: Dict[str, Any], additional: Optional[List[str]] = None
+) -> np.ndarray:
     """
     Renders text from the `info` dictionary to the `frame` image.
     """