def METHOD_NAME(self, dataset: FAIR1M) -> None:
    x = dataset[0].copy()
    dataset.plot(x, suptitle="Test")
    plt.close()
    dataset.plot(x, show_titles=False)
    plt.close()
    if dataset.split != "test":
        x["prediction_boxes"] = x["boxes"].clone()
        dataset.plot(x)
        plt.close()