
    hS                     T   S SK r S SKrS SKrS SKrS SKJrJrJrJrJ	r	J
r
Jr  S SKJr  S SKJrJr  S SKJr  SSKJr  SSKJr  \(       a  SS	KJr  / S
QrS\S\4S jr SS\S\S\S/S4   4S jjrS rS\	\\\\   4      4S jr   SS\S\S\	\\\\   4      S\S/S4   4S jjr g)    N)TYPE_CHECKINGCallableDictListOptionalSetType)with_nvtx_range)Modelwrap_model_recursive)use_nvtx_range   )Warnings)registry)Language)
pipepredictset_annotationsupdaterehearseget_loss
initializebegin_updatefinish_updater   forward_colorbackprop_colorc                 p   U R                    VVs/ s H-  u  p4[        US5      (       d  M  UR                  (       d  M+  UPM/     nnn[        5       nU HY  nUR                  R                  5        H8  n[        U5      U;   a  M  UR                  [        U5      5        [        XqUS9  M:     M[     U $ s  snnf )Nis_trainabler   r   )	
componentshasattrr   setmodelwalkidaddr
   )nlpr   r   _r   pipesseen_modelsnodes           L/home/james-whalen/.local/lib/python3.13/site-packages/spacy/ml/callbacks.pymodels_with_nvtx_ranger-   !   s     ~~%GA4( 	-1->-> 	% 
   EKJJOO%D$x;&OOBtH%.	 &  J!s   B2B2B2returnr   c                 6    [         R                  " [        U US9$ )Nr   )	functoolspartialr-   r   s     r,   create_models_with_nvtx_ranger2   5   s      #%     c                     [        U[        R                  5      (       a  U" U0 UD6$ [        U R                   SUR
                   35         U" U0 UD6sS S S 5        $ ! , (       d  f       g = f)N )
isinstancer0   r1   r   name__name__)selffuncargskwargss       r,   "nvtx_range_wrapper_for_pipe_methodr=   ?   sY    $	))**T$V$$tyyk4==/:;(( <;;s   A
A,additional_pipe_functionsc           
         U R                    H  u  p#U(       a  UR                  UR                  / 5      nO/ n[        U-    H  n[	        X5S 5      nUcB  XT;   a;  [
        R                  " [        R                  R                  XSR                  S95        MT  [        R                  " [        R                  " [        U5      U5      n [        R                   " U5      Ul         ['        UUU5        M     M     U $ !   [
        R                  " [        R$                  R                  XSR                  S95         M  = f! [(         a@    [
        R                  " [        R$                  R                  XSR                  S95         GMD  f = f)N)methodr   )r    getr7   %DEFAULT_NVTX_ANNOTATABLE_PIPE_METHODSgetattrwarningswarnr   W121formatr0   r1   types
MethodTyper=   inspect	signature__signature__W122setattrAttributeError)r'   r>   r(   r   extra_funcsr7   r:   wrapped_funcs           r,   pipes_with_nvtx_rangerR   G   s;    >>$377		2FKK9KGD4t,D|&MM(--"6"6d"6"ST$,,  !CTJDL-4->->t-D*Q + H "F Jhmm22$YY2OP " Qhmm22$YY2OPPQs   C9"D<9=D9<AFFc                    ^ ^^ UUU 4S jnU$ )Nc                 :   > [        U TT5      n [        U T5      n U $ )N)r-   rR   )r'   r>   r   r   s    r,   inner6create_models_and_pipes_with_nvtx_range.<locals>.inneru   s#    $S-H#C)BC
r3    )r   r   r>   rU   s   ``` r,   'create_models_and_pipes_with_nvtx_rangerX   p   s    

 Lr3   )rY   )rY   rY   N)!r0   rJ   rH   rD   typingr   r   r   r   r   r   r	   thinc.layersr
   thinc.modelr   r   
thinc.utilr   errorsr   utilr   languager   rB   intr-   r2   r=   strrR   rX   rW   r3   r,   <module>rc      s        K K K ( 3 %  #) %s C * 46-0zlJ&')&$,T#tCy.-A$B&T @D


  (S$s)^(<=
 zlJ&'	
r3   