U
    &ºc  ã                   @   sT   d dl mZmZmZ d dlZG dd„ deƒZG dd„ deƒZG dd„ deƒZd	Z	dS )
é    )ÚDictÚ
NamedTupleÚOptionalNc                   @   s&   e Zd ZU dZejed< ejed< dS )ÚEvalPredictionzX
    Evaluation output (always contains labels), to be used
    to compute metrics.
    ÚpredictionsÚ	label_idsN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚnpÚndarrayÚ__annotations__© r   r   ú>/tmp/pip-unpacked-wheel-ymerj3tt/transformers/trainer_utils.pyr      s   

r   c                   @   s:   e Zd ZU ejed< eej ed< eeee	f  ed< dS )ÚPredictionOutputr   r   ZmetricsN)
r   r	   r
   r   r   r   r   r   ÚstrÚfloatr   r   r   r   r      s   

r   c                   @   s   e Zd ZU eed< eed< dS )ÚTrainOutputZglobal_stepZtraining_lossN)r   r	   r
   Úintr   r   r   r   r   r   r      s   
r   Ú
checkpoint)
Útypingr   r   r   Znumpyr   r   r   r   ZPREFIX_CHECKPOINT_DIRr   r   r   r   Ú<module>   s
   
