def METHOD_NAME(self, raw: RawSegment) -> DepthInfo:
    """Get the depth info for a given segment."""
    try:
        return self.depth_info[raw.uuid]
    except KeyError as err:  # pragma: no cover
        reflow_logger.exception("Available UUIDS: %s", self.depth_info.keys())
        raise KeyError(
            "Tried to get depth info for unknown "
            f"segment {raw} with UUID {raw.uuid}"
        ) from err