code cleanup

rasa_nlu/extractors/entity_synonyms.py

@@ -1,22 +1,20 @@
-from __future__ import unicode_literals
-from __future__ import print_function
-from __future__ import division
 from __future__ import absolute_import
-from builtins import str
-from builtins import range
+from __future__ import division
+from __future__ import print_function
+from __future__ import unicode_literals
+
 import io
 import json
 import os
 import warnings
-import six
 
+import six
+from builtins import str
 from typing import Any
 from typing import Dict
-from typing import List
 from typing import Optional
 from typing import Text
 
-from rasa_nlu.config import RasaNLUConfig
 from rasa_nlu.extractors import EntityExtractor
 from rasa_nlu.model import Metadata
 from rasa_nlu.training_data import Message
@@ -41,7 +39,8 @@ def train(self, training_data, config, **kwargs):
         for example in training_data.entity_examples:
             for entity in example.get("entities", []):
                 entity_val = example.text[entity["start"]:entity["end"]]
-                self.add_entities_if_synonyms(entity_val, str(entity.get("value")))
+                self.add_entities_if_synonyms(entity_val,
+                                              str(entity.get("value")))
 
     def process(self, message, **kwargs):
         # type: (Message, **Any) -> None
@@ -54,9 +53,11 @@ def persist(self, model_dir):
         # type: (Text) -> Dict[Text, Any]
 
         if self.synonyms:
-            entity_synonyms_file = os.path.join(model_dir, "entity_synonyms.json")
+            entity_synonyms_file = os.path.join(model_dir,
+                                                "entity_synonyms.json")
             with io.open(entity_synonyms_file, 'w') as f:
-                f.write(str(json.dumps(self.synonyms, indent=2, separators=(',', ': '))))
+                f.write(str(json.dumps(self.synonyms, indent=2,
+                                       separators=(',', ': '))))
             return {"entity_synonyms": "entity_synonyms.json"}
         else:
             return {"entity_synonyms": None}