def METHOD_NAME(predictions: List[np.ndarray]) -> List[List[int]]:
    """Return a list containing the classes in batch."""
    return [bboxes_per_image.reshape((-1, 6))[:, -1].tolist() for bboxes_per_image in predictions]