catch correct exception type

Author: wochinge

rasa/nlu/featurizers/sparse_featurizer/count_vectors_featurizer.py

@@ -169,7 +169,7 @@ def _check_attribute_vocabulary(self, attribute: Text) -> bool:
         """Checks if trained vocabulary exists in attribute's count vectorizer."""
         try:
             return hasattr(self.vectorizers[attribute], "vocabulary_")
-        except (AttributeError, TypeError):
+        except (AttributeError, KeyError):
             return False
 
     def _get_attribute_vocabulary(self, attribute: Text) -> Optional[Dict[Text, int]]: