o
    
 ¢iñ  ã                   @   s(  d dl Z d dlmZ d dlmZmZmZ d dlmZ d dlmZ d dlm	Z	 e j
 dded	d	gd
dgddœeddddœfded	dgddgddœedddœfded	ddddœeddddœfded	ddddœedddœfde	dddddddœeddddœfde	dddddddœedddœfg¡dd„ ƒZdS )é    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           	      C   sŽ   | |d< | |d< t  d¡}t|di |¤Ž|di |¤Žƒ}| |¡ | |¡\}}t|ƒt|ƒks2J ‚|d jt|d ƒ| fksAJ ‚||ƒ d S )NÚwidthr   r   © )r   Ú	get_batchr   Ú
initializeÚbegin_updateÚlenÚshape)	r"   Ú
embed_archÚembed_configÚencode_archÚencode_configÚdocsÚtok2vecÚvectorsÚbackpropr#   r#   ú\/home/ubuntu/.local/lib/python3.10/site-packages/spacy_legacy/tests/pipeline/test_tok2vec.pyÚtest_tok2vec	   s   

r2   )ÚpytestÚspacy.testsr   Ú"spacy_legacy.architectures.tok2vecr   r   r   r   r   r   ÚmarkÚparametrizer2   r#   r#   r#   r1   Ú<module>   s"    $"" úý