
    h                     V   S SK r S SKJr  S SKJrJrJr  S SKJr  S SKJr  S SKJ	r	  \ R                  R                  SS\S	S	/S
S/SS.\SSSS.4S\S	S/SS/SS.\SSS.4S\S	SSSS.\SSSS.4S\S	SSSS.\SSS.4S\	SSSSSSS.\SSSS.4S\	SSSSSSS.\SSS.4/5      S 5       rg)    N)util)
Tok2Vec_v1MultiHashEmbed_v1CharacterEmbed_v1)MaxoutWindowEncoder_v1)MishWindowEncoder_v1)HashEmbedCNN_v1z7width,embed_arch,embed_config,encode_arch,encode_config   d   SHAPELOWERF)rowsattrsinclude_static_vectors         )window_sizemaxout_piecesdepth   ORTHPREFIX   )r   r   @   )r   nMnCr         T)r   
embed_sizer   r   subword_featurespretrained_vectorsc                 B   XS'   XS'   [         R                  " S5      n[        U" S0 UD6U" S0 UD65      nUR                  U5        UR	                  U5      u  px[        U5      [        U5      :X  d   eUS   R                  [        US   5      U 4:X  d   eU" U5        g )Nwidthr   r    )r   	get_batchr   
initializebegin_updatelenshape)	r$   
embed_archembed_configencode_archencode_configdocstok2vecvectorsbackprops	            b/home/james-whalen/.local/lib/python3.13/site-packages/spacy_legacy/tests/pipeline/test_tok2vec.pytest_tok2vecr4   	   s     ""'>>!D3l3[5Q=5QRGt,,T2Gw<3t9$$$1:DGe4444W    )pytestspacy.testsr   "spacy_legacy.architectures.tok2vecr   r   r   r   r   r	   markparametrizer4   r%   r5   r3   <module>r;      s     _ _ E C > = 
c
gw=Okpq  tJ  \]  pq  |}  L~  		
b	VX<Njop  sG  YZ  ef  Ig  	h	
2QZ_`bx  KL  _`  kl  {m  	n	
2QZ_`bv  IJ  UV  yW  	X	
OaQQWXlp  GL  M  Oe  wx  KL  WX  gY  	Z	
OaQQWXlp  GL  M  Oc  uv  AB  eC  	D		r5   