Skip to content

Commit 32c2ead

Browse files
committed
added random seed and number of epochs to crf extractor test
1 parent cb7d195 commit 32c2ead

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/nlu/extractors/test_crf_entity_extractor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ def test_crf_extractor(spacy_nlp, ner_crf_pos_feature_config):
1313
"entities": [
1414
{"start": 16, "end": 20, "value": "west", "entity": "location"}
1515
],
16-
"spacy_doc": spacy_nlp("anywhere in the west"),
16+
SPACY_DOCS[TEXT]: spacy_nlp("anywhere in the west"),
1717
},
1818
),
1919
Message(
@@ -36,7 +36,7 @@ def test_crf_extractor(spacy_nlp, ner_crf_pos_feature_config):
3636
"extractor": "CRFEntityExtractor",
3737
},
3838
],
39-
"spacy_doc": spacy_nlp("central indian restaurant"),
39+
SPACY_DOCS[TEXT]: spacy_nlp("central indian restaurant"),
4040
},
4141
),
4242
]

0 commit comments

Comments
 (0)