From 7ef234659810a8c18c0f3fc60e957b40b4cd5c35 Mon Sep 17 00:00:00 2001 From: envolution Date: Sat, 30 Nov 2024 15:23:31 -0500 Subject: [PATCH] try fix gil require issue --- spacy/ml/parser_model.pyx | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/spacy/ml/parser_model.pyx b/spacy/ml/parser_model.pyx index 8698fc5b6..6a3aae915 100644 --- a/spacy/ml/parser_model.pyx +++ b/spacy/ml/parser_model.pyx @@ -527,7 +527,7 @@ 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(float[:] scores, const int n_classes) nogil: if n_classes == 2: return 0 if scores[0] > scores[1] else 1 cdef int i @@ -537,4 +537,5 @@ cdef class precompute_hiddens: if scores[i] > mode: mode = scores[i] best = i - return best + return best +