def METHOD_NAME(
    self,
    sample: dict[str, Tensor],
    show_titles: bool = True,
    suptitle: Optional[str] = None,
) -> plt.Figure:
    """Plot a sample from the dataset.
    Args:
        sample: a sample returned by :meth:`NonGeoClassificationDataset.__getitem__`
        show_titles: flag indicating whether to show titles above each panel
        suptitle: optional suptitle to use for figure
    Returns:
        a matplotlib Figure with the rendered sample
    .. versionadded:: 0.2
    """
    image, label = sample["image"], cast(int, sample["label"].item())
    showing_predictions = "prediction" in sample
    if showing_predictions:
        prediction = cast(int, sample["prediction"].item())
    fig, ax = plt.subplots(1, 1, figsize=(10, 10))
    ax.imshow(image.permute(1, 2, 0))
    ax.axis("off")
    if show_titles:
        title = f"Label: {self.classes[label]}"
        if showing_predictions:
            title += f"\nPrediction: {self.classes[prediction]}"
        ax.set_title(title)
    if suptitle is not None:
        plt.suptitle(suptitle)
    return fig