Skip to content

Commit a0cdc2b

Browse files
authored
Use Language.pipe in evaluate (explosion#9800)
1 parent 9964243 commit a0cdc2b

File tree

1 file changed

+7
-14
lines changed

1 file changed

+7
-14
lines changed

spacy/language.py

Lines changed: 7 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1404,20 +1404,13 @@ def evaluate(
14041404
for eg in examples:
14051405
self.make_doc(eg.reference.text)
14061406
# apply all pipeline components
1407-
for name, pipe in self.pipeline:
1408-
kwargs = component_cfg.get(name, {})
1409-
kwargs.setdefault("batch_size", batch_size)
1410-
for doc, eg in zip(
1411-
_pipe(
1412-
(eg.predicted for eg in examples),
1413-
proc=pipe,
1414-
name=name,
1415-
default_error_handler=self.default_error_handler,
1416-
kwargs=kwargs,
1417-
),
1418-
examples,
1419-
):
1420-
eg.predicted = doc
1407+
docs = self.pipe(
1408+
(eg.predicted for eg in examples),
1409+
batch_size=batch_size,
1410+
component_cfg=component_cfg,
1411+
)
1412+
for eg, doc in zip(examples, docs):
1413+
eg.predicted = doc
14211414
end_time = timer()
14221415
results = scorer.score(examples)
14231416
n_words = sum(len(eg.predicted) for eg in examples)

0 commit comments

Comments
 (0)