Merge pull request #46 from zalandoresearch/GH-23-retagging-error

GH-23: retagging error

flair/models/sequence_tagger_model.py

@@ -11,6 +11,8 @@
 
 from typing import List, Tuple, Union
 
+from flair.training_utils import clear_embeddings
+
 START_TAG: str = '<START>'
 STOP_TAG: str = '<STOP>'
 
@@ -359,6 +361,9 @@ def predict(self, sentences: Union[List[Sentence], Sentence], mini_batch_size=32
         if type(sentences) is Sentence:
             sentences = [sentences]
 
+        # remove previous embeddings
+        clear_embeddings(sentences)
+
         # make mini-batches
         batches = [sentences[x:x + mini_batch_size] for x in range(0, len(sentences), mini_batch_size)]
 

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name='flair',
-    version='0.2.0',
+    version='0.2.0.post1',
     description='A very simple framework for state-of-the-art NLP',
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",

tests/test_sequence_tagger.py

@@ -0,0 +1,17 @@
+from flair.data import Sentence
+from flair.models import SequenceTagger
+
+
+def test_tag_sentence():
+
+    # test tagging
+    sentence = Sentence('I love Berlin')
+
+    tagger = SequenceTagger.load('ner')
+
+    tagger.predict(sentence)
+
+    # test re-tagging
+    tagger = SequenceTagger.load('pos')
+
+    tagger.predict(sentence)