From 70b2d19423626c1e9550d6d9cd43cb94e866b7d9 Mon Sep 17 00:00:00 2001 From: Eliana Vornov Date: Wed, 24 Jan 2024 10:05:05 -0500 Subject: [PATCH] better copying for warmup docs --- spacy/cli/benchmark_speed.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/spacy/cli/benchmark_speed.py b/spacy/cli/benchmark_speed.py index 41dbcd455..4dd10049c 100644 --- a/spacy/cli/benchmark_speed.py +++ b/spacy/cli/benchmark_speed.py @@ -173,7 +173,5 @@ def print_outliers(sample: numpy.ndarray): def warmup( nlp: Language, docs: List[Doc], warmup_epochs: int, batch_size: Optional[int] ) -> numpy.ndarray: - warmup_docs = [] - for _ in range(warmup_epochs): - warmup_docs += [doc.from_docs([doc]) for doc in docs] - return annotate(nlp, warmup_docs, batch_size) + docs = [doc.copy() for doc in docs * warmup_epochs] + return annotate(nlp, docs, batch_size)