def METHOD_NAME(self, dataset: IDTReeS) -> None:
    x = dataset[0].copy()
    dataset.plot(x, suptitle="Test")
    plt.close()
    dataset.plot(x, show_titles=False)
    plt.close()
    if "boxes" in x:
        x["prediction_boxes"] = x["boxes"]
        dataset.plot(x, show_titles=True)
        plt.close()
    if "label" in x:
        x["prediction_label"] = x["label"]
        dataset.plot(x, show_titles=False)
        plt.close()