Fix test

2017-04-24 00:25:55 +02:00 · 2017-04-24 00:25:55 +02:00 · 65f10b53e5
parent 70a43858e1
commit 65f10b53e5
1 changed files with 3 additions and 1 deletions
--- a/spacy/tests/regression/test_issue999.py
+++ b/spacy/tests/regression/test_issue999.py
@ -56,6 +56,7 @@ def test_issue999(train_data):
    for _, offsets in train_data:
        for start, end, ent_type in offsets:
            nlp.entity.add_label(ent_type)
+    nlp.entity.model.learn_rate = 0.001
    for itn in range(100):
        random.shuffle(train_data)
        for raw_text, entity_offsets in train_data:
@ -74,4 +75,5 @@ def test_issue999(train_data):
                assert ents[(start, end)] == label
                break
        else:
-            raise Exception(ents)
+            if entity_offsets:
+                raise Exception(ents)