diff --git a/predicators/envs/spot_env.py b/predicators/envs/spot_env.py index c7a1f22fef..06bc0bd971 100644 --- a/predicators/envs/spot_env.py +++ b/predicators/envs/spot_env.py @@ -324,8 +324,8 @@ def _build_observation(self, object_names_in_view_by_camera_apriltag) # Additionally, if we're using SAM, then update using that. if CFG.spot_grasp_use_sam: - object_names_in_view_by_camera_sam = self._spot_interface.get_objects_in_view_by_camera( - from_apriltag=True) + object_names_in_view_by_camera_sam = self._spot_interface.\ + get_objects_in_view_by_camera(from_apriltag=True) # Combine these together to get all objects in view. for k, v in object_names_in_view_by_camera.items(): v.update(object_names_in_view_by_camera_sam[k]) diff --git a/predicators/spot_utils/spot_utils.py b/predicators/spot_utils/spot_utils.py index d30d6dea06..3cf8a4a2ac 100644 --- a/predicators/spot_utils/spot_utils.py +++ b/predicators/spot_utils/spot_utils.py @@ -273,7 +273,6 @@ def get_camera_images(self) -> Dict[str, Image]: """Get all camera images.""" camera_images: Dict[str, Image] = {} for source_name in CAMERA_NAMES: - # TODO: convert to RGB - should be correct img, _ = self.get_single_camera_image(source_name, to_rgb=True) camera_images[source_name] = img return camera_images