o
    i                     @   sp   d dl Z d dlmZmZmZmZmZmZmZm	Z	 d dl
mZmZ e jjdddd Ze jjdddd	 ZdS )
    N)ParametricAttentionReluSoftmaxchainlist2ragged
reduce_max
reduce_sumwith_ragged)DataValidationErrordata_validationz5Validation currently disabled for Pydantic 2 changes0)reasonc               	   C   sD  t tdtdtt t } td tt | j	| j
dd| j
ddd W d    n1 s5w   Y  tt | j	| j
ddd| j
ddd W d    n1 s\w   Y  tt | j	| j
ddg| j
ddd W d    n1 sw   Y  W d    d S W d    d S 1 sw   Y  d S )N
   T   )XY)r   r   r	   r   r   r   pytestraisesr
   
initializeopsalloc2falloc3f)model r   U/home/ubuntu/.local/lib/python3.10/site-packages/thinc/tests/model/test_validation.pytest_validation   s   
$&&"r   c               	   C   s   t t t tdddtd} | jjjdddg}| jjjddd}| || | | t t t tdddt	dtd}t
d	, tt ||| W d    n1 sZw   Y  W d    d S W d    d S 1 srw   Y  d S )
N   g      ?)dropoutr   )   K   f)dtype)r   T)r   r   r   r   r   xpzerosr   predictr   r   r   r   r
   )
good_modelr   r   	bad_modelr   r   r   test_validation_complex   s&   


"r&   )r   	thinc.apir   r   r   r   r   r   r   r	   
thinc.utilr
   r   markxfailr   r&   r   r   r   r   <module>   s    (

