o
    soi                     @  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dS )NonZeroreturnNonec                  C  s^   t jjddgdgd} tjddgddggtd}tjt|tjd}t| |g|gdd	 d S )
Nr   	conditionresult)inputsoutputs   r   )dtypetest_nonzero_example)r
   r   name)	onnxhelper	make_nodenparrayboolnonzeroint64r   )noder   r	    r   W/home/ubuntu/.local/lib/python3.10/site-packages/onnx/backend/test/case/node/nonzero.pyexport   s   zNonZero.exportN)r   r   )__name__
__module____qualname__staticmethodr   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   