added typing to overlay_frame

bdaiinstitute · Apr 2, 2024 · 1de011b · 1de011b
1 parent d79d7f1
commit 1de011b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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.
     """