o
    wi                     @  sH   d dl mZ d dlZd dlZd dlmZ d dlmZ G dd deZ	dS )    )annotationsN)Base)expectc                   @  s(   e Zd ZedddZedddZdS )	IsNaNreturnNonec                  C  ^   t jjddgdgd} tjdtjtjdtj tjgtjd}t|}t	| |g|gdd	 d S )
Nr   xyinputsoutputs333333ffffff@dtype
test_isnanr   r   name)
onnxhelper	make_nodenparraynaninffloat32isnanr   noder	   r
    r    ^/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/isnan.pyexport      &
zIsNaN.exportc                  C  r   )
Nr   r	   r
   r   r   r   r   test_isnan_float16r   )
r   r   r   r   r   r   r   float16r   r   r   r    r    r!   export_float16   r#   zIsNaN.export_float16N)r   r   )__name__
__module____qualname__staticmethodr"   r&   r    r    r    r!   r      s
    r   )

__future__r   numpyr   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r    r    r    r!   <module>   s   