
    h                     ,   S SK JrJrJrJrJr  SSKJr  \" S5      r\" S5      r	S\\\	4   S\\\   \\	   4   4S jr
S	\\\   \\	   4   S
\\   S\S\\\	   \\\	   /\\   4   4   4S jr  SS	\\\   \\	   4   S\\\      S\\\	      SS4S jjrg)    )CallableListOptionalTupleTypeVar   )ModelInTOutTlayerreturnc                 ,    [        S[        U /[        S9$ )z:Create a model that maps a child layer across list inputs.map_list)layersinit)r	   forwardr   )r   s    O/home/james-whalen/.local/lib/python3.13/site-packages/thinc/layers/map_list.pyr   r   	   s    WeW4@@    modelXsis_trainc                    ^	 U R                   S   n/ n/ m	U H/  nU" XR5      u  pgUR                  U5        T	R                  U5        M1     S[        [           S[        [           4U	4S jjnXH4$ )Nr   dYsr   c                 ^   > [        TU 5       VVs/ s H  u  pU" U5      PM     snn$ s  snnf )N)zip)r   callbackdY	callbackss      r   backprop_map_list"forward.<locals>.backprop_map_list   s*    14Y1DE1D1DEEEs   ))r   appendr   r   r
   )
r   r   r   r   YsXYget_dXr   r   s
            @r   r   r      st     LLOE	BI!&	
		!  
FtDz Fd3i F   r   Nr#   r$   c                 n    U R                   S   R                  U(       a  US   OS U(       a  US   OS S9  g )Nr   )r#   r$   )r   
initialize)r   r#   r$   s      r   r   r      s/    
 
LLO1!tLr   )NN)typingr   r   r   r   r   r   r	   r
   r   r   boolr   r    r   r   <module>r+      s   ; ; envAE#t)$ AtCy$t*/D)E A
!cDJ&'!-1#Y!BF!
4:xdd3i 7889!& ""McDJ&'MS	M T
M 
	Mr   