diff --git a/rasa/nlu/emulators/luis.py b/rasa/nlu/emulators/luis.py index b2653548cb65..4c5b5975096d 100644 --- a/rasa/nlu/emulators/luis.py +++ b/rasa/nlu/emulators/luis.py @@ -37,13 +37,11 @@ def _intents(self, data: Dict[Text, Any]) -> Dict[Text, Any]: return {top[INTENT_NAME_KEY]: {"score": top[PREDICTED_CONFIDENCE_KEY]}} - def _entities( - self, data: Dict[Text, Any] - ) -> Dict[Text, Dict[Text, List[Dict[Text, Any]]]]: + def _entities(self, data: Dict[Text, Any]) -> Dict[Text, Any]: if ENTITIES not in data: return {} - entities: Dict[Text, Dict[Text, List[Dict[Text, Any]]]] = {"$instance": {}} + entities: Dict[Text, Any] = {"$instance": {}} for e in data[ENTITIES]: # LUIS API v3 uses entity roles instead of entity names # (it's possible because its roles are unique): diff --git a/rasa/nlu/utils/hugging_face/transformers_pre_post_processors.py b/rasa/nlu/utils/hugging_face/transformers_pre_post_processors.py index a716ebb78658..232b89f4d571 100644 --- a/rasa/nlu/utils/hugging_face/transformers_pre_post_processors.py +++ b/rasa/nlu/utils/hugging_face/transformers_pre_post_processors.py @@ -22,11 +22,10 @@ def cleanup_tokens( # remove empty strings token_ids_string = [(id, string) for id, string in token_ids_string if string] - token_ids: List[int] - token_strings: List[Text] # return as individual token ids and token strings token_ids, token_strings = zip(*token_ids_string) - return token_ids, token_strings + # FIXME: zip official typing is not really properly set up + return token_ids, token_strings # type: ignore[return-value] def bert_tokens_pre_processor(token_ids: List[int]) -> List[int]: