
    h              
           S SK JrJrJrJr  SSKJr  SSKJr  SSK	J
r
Jr  \r\" S\
S9r\R                  " S5      S	\\\4   4S
 j5       rS\\\4   S\S\S	\\\4   4S jrg)    )CallableTupleTypeVarcast   )registry)Model)ListXdRaggedOutT)boundzragged2list.v1returnc                  "    [        S[        5      $ )z4Transform sequences from a ragged format into lists.ragged2list)r	   forward     R/home/james-whalen/.local/lib/python3.13/site-packages/thinc/layers/ragged2list.pyr   r      s     ((r   modelXris_trainc                    ^ ^ UR                   mS[        S[        4UU 4S jjn[        [        T R                  R                  UR                  UR                   5      5      nXC4$ )NdXsr   c                 L   > [        TR                  R                  U SS9T5      $ )Nr   )pad)r   opsflatten)r   lengthsr   s    r   backpropforward.<locals>.backprop   s$    eii'''3W==r   )r   r   InTr   r   	unflattendataXd)r   r   r   r   datar   s   `    @r   r   r      sS    jjG>d >s > > eii))"))RZZ@AD>r   N)typingr   r   r   r   configr   r   r	   typesr
   r   r!   r   layersr   boolr   r   r   r   <module>r*      s    1 1   "vV$ 
!")U39% ) #)
5d#   tX~AV r   