From afe6ee4548a76cdbad03b4949bb88cc6d6468bf3 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Mon, 22 Jun 2020 16:28:47 +0200 Subject: [PATCH] Fix Corpus --- spacy/gold/corpus.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/spacy/gold/corpus.py b/spacy/gold/corpus.py index 1244e2516..3318a7eef 100644 --- a/spacy/gold/corpus.py +++ b/spacy/gold/corpus.py @@ -72,7 +72,7 @@ class Corpus: i += 1 return n - def train_dataset(self, nlp, shuffle=True): + def train_dataset(self, nlp, shuffle=True, **kwargs): ref_docs = self.read_docbin(nlp.vocab, self.walk_corpus(self.train_loc)) examples = self.make_examples(nlp, ref_docs) if shuffle: @@ -80,7 +80,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.dev_loc)) examples = self.make_examples(nlp, ref_docs) yield from examples