From e67cf1a5a548e18d98297981551100c1663bf7eb Mon Sep 17 00:00:00 2001
From: envolution <involution@gmail.com>
Date: Sat, 30 Nov 2024 15:16:43 -0500
Subject: [PATCH] fix paste error

---
 spacy/ml/parser_model.pyx | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/spacy/ml/parser_model.pyx b/spacy/ml/parser_model.pyx
index 7375375a7..8698fc5b6 100644
--- a/spacy/ml/parser_model.pyx
+++ b/spacy/ml/parser_model.pyx
@@ -527,4 +527,14 @@ cdef class precompute_hiddens:
 
         return state_vector, backprop_relu
 
-cdef inline int _arg_max(const float* scores, const int n_classes) nogil:
+    cdef inline int _arg_max(const float* scores, const int n_classes) nogil:
+        if n_classes == 2:
+            return 0 if scores[0] > scores[1] else 1
+        cdef int i
+        cdef int best = 0
+        cdef float mode = scores[0]
+        for i in range(1, n_classes):
+            if scores[i] > mode:
+                mode = scores[i]
+                best = i
+        return best