From a9ed8bb4013caeb1a45c282ac38d47207362d525 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sat, 14 Sep 2024 12:55:53 +0200 Subject: [PATCH] Replace numpy floats in evaluate and update --- spacy/language.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/spacy/language.py b/spacy/language.py index 0d9aab9e3..18a58afd5 100644 --- a/spacy/language.py +++ b/spacy/language.py @@ -1212,7 +1212,7 @@ class Language: examples, ): eg.predicted = doc - return losses + return _replace_numpy_floats(losses) def rehearse( self, @@ -1463,7 +1463,7 @@ class Language: results = scorer.score(examples, per_component=per_component) n_words = sum(len(eg.predicted) for eg in examples) results["speed"] = n_words / (end_time - start_time) - return results + return _replace_numpy_floats(results) def create_optimizer(self): """Create an optimizer, usually using the [training.optimizer] config."""