
    Mi                     4   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
  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)with_nvtx_range)use_nvtx_range   )Warnings)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/var/www/eduai.edurigo.com/storigo/production/storigo_env/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_ranger.   3   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 )
isinstancer,   r-   r
   name__name__)selffuncargskwargss       r(   "nvtx_range_wrapper_for_pipe_methodr9   =   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   getr3   %DEFAULT_NVTX_ANNOTATABLE_PIPE_METHODSgetattrwarningswarnr   W121formatr,   r-   types
MethodTyper9   inspect	signature__signature__W122setattrAttributeError)r#   r:   r$   r   extra_funcsr3   r6   wrapped_funcs           r(   pipes_with_nvtx_rangerN   E   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)   rN   )r#   r:   r   r   s    r(   inner6create_models_and_pipes_with_nvtx_range.<locals>.inners   s#    $S-H#C)BC
r/    )r   r   r:   rQ   s   ``` r(   'create_models_and_pipes_with_nvtx_rangerT   n   s    

 Lr/   )rU   )rU   rU   N)r,   rF   rD   r@   typingr   r   r   r   r   r   thinc.layersr	   
thinc.utilr
   errorsr   languager   r>   intr)   r.   r9   strrN   rT   rS   r/   r(   <module>r]      s        E E ( % #) %s C * 46-0zlJ&')&$,T#tCy.-A$B&T @D


  (S$s)^(<=
 zlJ&'	
r/   