o
    }o™i–  ã                   @   s,   d dl mZmZ d dlZG dd„ deƒZdS )é    )ÚABCÚabstractmethodNc                   @   s8   e Zd Zedd„ ƒZedd„ ƒZedejfdd„ƒZdS )	ÚITritonDeployablec                 C   ó   d S ©N© ©Úselfr   r   úQ/home/ubuntu/.local/lib/python3.10/site-packages/nemo/deploy/triton_deployable.pyÚget_triton_input   ó   z"ITritonDeployable.get_triton_inputc                 C   r   r   r   r   r   r   r
   Úget_triton_output   r   z#ITritonDeployable.get_triton_outputÚinputsc                 K   r   r   r   )r	   r   r   r   r
   Útriton_infer_fn   r   z!ITritonDeployable.triton_infer_fnN)	Ú__name__Ú
__module__Ú__qualname__r   r   r   ÚnpÚndarrayr   r   r   r   r
   r      s    

r   )Úabcr   r   Únumpyr   r   r   r   r   r
   Ú<module>   s   