o
    i                  	   @   s   d dl Z d dlZd dlmZ d dlmZmZmZmZm	Z	 ej
dee	eegdd Zej
ddi fd	i fd
dddfdddifgdd ZdS )    N)registry)NumpyOpsglorot_uniform_initnormal_inituniform_init	zero_init	init_funcc                 C   s8   t  }tjg ddd}| ||j}t||rJ d S )N            f)dtype)r   numpyndarrayshapearray_equal)r   opsdataresult r   Q/home/ubuntu/.local/lib/python3.10/site-packages/thinc/tests/test_initializers.pytest_initializer_func_setup   s   r   zname,kwargszglorot_uniform_init.v1zzero_init.v1zuniform_init.v1g      g      ?)lohiznormal_init.v1meang?c                 C   s.   dd| i|i}t |d }|t d dS )z[Test that initializers are loaded and configured correctly from registry
    (as partials).testz@initializersr	   N)r   resolver   )namekwargscfgfuncr   r   r   test_initializer_from_config   s   r#   )r   pytestthincr   	thinc.apir   r   r   r   r   markparametrizer   r#   r   r   r   r   <module>   s"    	

	