
    h                         S SK JrJrJr  S SKJrJrJr  S SKJ	r	J
r
  SSKJr  SSKJr  \\   r\
rS\\\\
   4   S\\\4   S	\\\4   4S
 jrS	\\\
   \
4   4S jrg)    )CallableListTuple)Modelchain
with_array)Floats1dFloats2d   )Doc)registrytok2vecscorerreturnc                     [        U5      n[        X[        5       5      nUR                  SU 5        UR                  SU5        UR                  SU5        U$ )Nr   r   logistic_layer)r   r   	flattenerset_ref)r   r   r   models       U/home/james-whalen/.local/lib/python3.13/site-packages/spacy/ml/models/span_finder.pybuild_finder_modelr      sM     =Gv<NN#GY[IE	MM)W%	MM(F#	MM"N3L    c                      S[         [        [        4   S[        [           S[        S[
        [        [        [        /[        [           4   4   4S jn [        SU S9$ )z4Flattens the input to a 1-dimensional list of scoresr   Xis_trainr   c                    ^ ^ T R                   R                  U Vs/ s H  n[        U5      PM     sn5      mT R                   R                  U5      nS[        S[
        [           4UU 4S jjnXE4$ s  snf )NdYr   c                 <   > TR                   R                  U T5      $ )N)ops	unflatten)r   lensr   s    r   backprop,flattener.<locals>.forward.<locals>.backprop#   s    99&&r400r   )r   	asarray1ilenflattenr
   r   )r   r   r   docYr"   r!   s   `     @r   forwardflattener.<locals>.forward   sk     yy""#:CH#:;IIa 	1 	1d8n 	1 	1 { $;s   A2	Flattenerr)   )r   r	   r   r
   boolr   r   r,   s    r   r   r      s^    	Xx'(	-1(^	GK		x8*d8n"<==	>	 g..r   N)typingr   r   r   	thinc.apir   r   r   thinc.typesr	   r
   tokensr   utilr   InTOutTr   r    r   r   <module>r6      s    ( ( . . *  
3i
3X&'
16tTz1B

39
/5h12 /r   