From 7d329cd1acf0c8ee11b76d6222bacaceb16328b8 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sun, 21 Jun 2020 17:16:55 +0200 Subject: [PATCH] Add kwargs to Corpus.dev_dataset to match train_dataset --- spacy/gold/corpus.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/gold/corpus.py b/spacy/gold/corpus.py index 750217c8c..14774354c 100644 --- a/spacy/gold/corpus.py +++ b/spacy/gold/corpus.py @@ -76,7 +76,7 @@ class Corpus: random.shuffle(examples) yield from examples - def dev_dataset(self, nlp): + def dev_dataset(self, nlp, **kwargs): ref_docs = self.read_docbin(nlp.vocab, self.walk_corpus(self.train_loc)) examples = self.make_examples(nlp, ref_docs, **kwargs) yield from examples