def METHOD_NAME(self, predictions):
    if self._enabled:
        dataset = self.get_dataset("train")
        viz = Visualizer()
        lut = LabelLUT()
        for id_color, color in dataset.color_map.items():
            lut.add_label(str(id_color), id_color, color=color)
        viz.set_lut("label", lut)
        for prediction_batch in predictions:
            for pred in prediction_batch:
                data = {
                    "points": pred[DataKeys.INPUT][:, :3],
                    "name": pred[DataKeys.METADATA],
                }
                bounding_box = pred[DataKeys.PREDS]
                viz.visualize([data], bounding_boxes=bounding_box)