@staticmethod
def _check_enough_labels(model_data: RasaModelData) -> bool:
    return (len(np.unique(model_data.get(LABEL_KEY, LABEL_SUB_KEY))) >= 2)
@staticmethod
def _check_enough_labels(model_data: RasaModelData) -> bool:
    return (len(np.unique(model_data.get(LABEL_KEY, LABEL_SUB_KEY))) >= 2)