o
    ij                     @   sV   d dl mZ d dlZd dlmZmZ d dlmZmZ ej	dd Z
dd Zd	d
 ZdS )    )partialN)Linear	resizable)resize_linear_weightedresize_modelc                  C   s.   t d d d} ddd}t| tt|dd}|S )N)nOnIr   )bW)fill_defaults)resize_layer)r   r   r   r   )output_layerr   model r   U/home/ubuntu/.local/lib/python3.10/site-packages/thinc/tests/layers/test_resizable.pyr   	   s   

r   c                 C   s   | j dksJ d S )Nzresizable(linear))namer   r   r   r   "test_resizable_linear_default_name   s   r   c                 C   sd   t | dd t | dd | jdddd t | dd | jdddd | jdddd t | dd d	S )
z8Test that resizing the model doesn't cause an exception.
   )new_nO   r   r   T)forcer   N)r   set_dimr   r   r   r   test_resize_model   s   r   )	functoolsr   pytest	thinc.apir   r   thinc.layers.resizabler   r   fixturer   r   r   r   r   r   r   <module>   s    

