fix remaining issues in rasa.nlu

RasaHQ · Mar 10, 2022 · 4e5e82d · 4e5e82d
1 parent 7b418ab
commit 4e5e82d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git 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
@@ -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]: