Ë
    `›¾jý  ã                   ó  — d dl Z d dlZd dlZd dlZd dlmZmZmZmZm	Z	m
Z
 d dlmZ d dlmZ ddlmZ erddlmZ g d¢Zd	ed
efd„Z	 dd	ed
ededgdf   fd„Zd„ Zde	eeee   f      fd„Z	 	 	 dd	ed
ede	eeee   f      dedgdf   fd„Zy)é    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                 óJ  — | j                   D ��cg c]  \  }}t        |d«      r|j                  r|‘Œ! }}}t        «       }|D ]W  }|j                  j                  «       D ]8  }t        |«      |v rŒ|j                  t        |«      «       t        |||¬«       Œ: ŒY | S c c}}w )NÚis_trainable©r   r   )	Ú
componentsÚhasattrr   ÚsetÚmodelÚwalkÚidÚaddr	   )Únlpr   r   Ú_r   ÚpipesÚseen_modelsÚnodes           ú\/home/spazio/kingdom_calls/work5/transenv/lib/python3.12/site-packages/spacy/ml/callbacks.pyÚmodels_with_nvtx_ranger)      s§   € ð —~‘~÷áˆAˆtÜ�4˜Ô(¨T×->Ò->ò 	ð€Eñ ô  ›E€KØò ˆØ—J‘J—O‘OÓ%ò 	ˆDÜ�$‹x˜;Ñ&ØØ�O‰OœB˜t›HÔ%ÜØ MÀ.öñ		ðð €Jùó!s   �$BÚreturnr   c                 ó:   — t        j                  t        | |¬«      S )Nr   )Ú	functoolsÚpartialr)   r   s     r(   Úcreate_models_with_nvtx_ranger.   3   s    € ô ×ÑÜØ#Ø%ôð ó    c                 óÊ   — t        |t        j                  «      r ||i |¤ŽS t        | j                  › d|j
                  › �«      5   ||i |¤Žcd d d «       S # 1 sw Y   y xY w)Nú )Ú
isinstancer,   r-   r
   ÚnameÚ__name__)ÚselfÚfuncÚargsÚkwargss       r(   Ú"nvtx_range_wrapper_for_pipe_methodr9   =   s_   € Ü�$œ	×)Ñ)Ô*Ù�TÐ$˜VÑ$Ð$ä˜tŸy™y˜k¨¨4¯=©=¨/Ð:Ó;ñ 	)Ù˜Ð( Ñ(÷	)÷ 	)ò 	)ús   ÁAÁA"Úadditional_pipe_functionsc           	      ó  — | j                   D ]Ý  \  }}|r|j                  |j                  g «      }ng }t        |z   D ]«  }t	        ||d «      }|€C||v r>t        j                  t        j                  j                  ||j                  ¬«      «       ŒUt        j                  t        j                  t        |«      |«      }	 t        j                   |«      |_        	 t'        |||«       Œ­ Œß | S #  t        j                  t        j$                  j                  ||j                  ¬«      «       Y ŒôxY w# t(        $ rB t        j                  t        j$                  j                  ||j                  ¬«      «       Y �ŒAw xY w)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   sQ  € ð —>‘>ò !Q‰ˆˆ4Ù$Ø3×7Ñ7¸¿	¹	À2ÓF‰KàˆKä9¸KÑGò 	QˆDÜ˜4  tÓ,ˆDØˆ|Ø˜;Ñ&Ü—M‘M¤(§-¡-×"6Ñ"6¸dÈÏÉÐ"6Ó"SÔTØä$×,Ñ,Ü× Ñ Ô!CÀTÓJÈDóˆLðÜ-4×->Ñ->¸tÓ-D�Ô*ðQÜØØØ õñ+	Qð!QðF €Jøðä—‘œhŸm™m×2Ñ2¸$ÀTÇYÁYÐ2ÓOÔPÙûô "ò QÜ—‘œhŸm™m×2Ñ2¸$ÀTÇYÁYÐ2ÓO×PðQús    ÃC/ÃD4Ã/A D1Ä4AE?Å>E?c                 ó   ‡ ‡‡— ˆˆˆ fd„}|S )Nc                 ó:   •— t        | ‰‰«      } t        | ‰«      } | S )N)r)   rN   )r#   r:   r   r   s    €€€r(   Úinnerz6create_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_rangerS   n   s   ú€ ö
ð
 €Lr/   )éÿÿÿÿrT   )rT   rT   N)r,   rF   rD   r@   Útypingr   r   r   r   r   r   Úthinc.layersr	   Ú
thinc.utilr
   Úerrorsr   Úlanguager   r>   Úintr)   r.   r9   ÚstrrN   rS   rR   r/   r(   ú<module>r\      sæ   ðÛ Û Û Û ß E× Eå (Ý %å áå#ò)Ð %ð¨sð ÀCó ð* 46ñØðØ-0ðàˆzˆl˜JÐ&Ñ'óò)ð&Ø$,¨T°#°t¸C±y°.Ñ-AÑ$Bó&ðT ØØ@Dñ
Øð
àð
ð  (¨¨S°$°s±)¨^Ñ(<Ñ=ð
ð ˆzˆl˜JÐ&Ñ'ô	
r/   