o
    Xεi
                     @   sL   d dl mZmZ d dlmZmZ d dlmZ d dlm	Z	 G dd deZ
dS )    )ABCabstractmethod)DictUnionsave_fsspec)rank_zero_onlyc                   @   s  e Zd ZededededdfddZededed	 deddfd
dZ	edd Z
ededddededdf
ddZededededdfddZed?dededefddZedededefddZedededefd d!Zeded"ededefd#d$Zed%d& Zed'd( Zeed)ed*efd+d,Zd-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Zd=d> ZdS )@BaseDashboardLoggertitlevaluestepreturnNc                 C      d S N )selfr
   r   r   r   r   T/home/ubuntu/.local/lib/python3.10/site-packages/trainer/logging/base_dash_logger.py
add_scalar
      zBaseDashboardLogger.add_scalarfigure)zmatplotlib.figure.Figurezplotly.graph_objects.Figurec                 C   r   r   r   )r   r
   r   r   r   r   r   
add_figure   s   zBaseDashboardLogger.add_figurec                 C   r   r   r   )r   configr   r   r   
add_config   r   zBaseDashboardLogger.add_configaudioz
np.ndarraysample_ratec                 C   r   r   r   )r   r
   r   r   r   r   r   r   	add_audio   r   zBaseDashboardLogger.add_audiotextc                 C   r   r   r   )r   r
   r   r   r   r   r   add_text   r   zBaseDashboardLogger.add_textfile_or_dirnameartifact_typec                 C   r   r   r   )r   r   r   r    aliasesr   r   r   add_artifact#   r   z BaseDashboardLogger.add_artifact
scope_namescalarsc                 C   r   r   r   )r   r#   r$   r   r   r   r   add_scalars'   r   zBaseDashboardLogger.add_scalarsfiguresc                 C   r   r   r   )r   r#   r&   r   r   r   r   add_figures+   r   zBaseDashboardLogger.add_figuresaudiosc                 C   r   r   r   )r   r#   r(   r   r   r   r   r   
add_audios/   r   zBaseDashboardLogger.add_audiosc                 C   r   r   r   r   r   r   r   flush3   r   zBaseDashboardLogger.flushc                 C   r   r   r   r*   r   r   r   finish7   r   zBaseDashboardLogger.finishstatepathc                 C   s   t | | d S r   r   )r-   r.   r   r   r   
save_model;   s   zBaseDashboardLogger.save_modelc                 C      | j d||d d S )NTrainIterStatsr#   r$   r   r%   r   r   statsr   r   r   train_step_stats@      z$BaseDashboardLogger.train_step_statsc                 C   r0   )NTrainEpochStatsr2   r3   r4   r   r   r   train_epoch_statsC   r7   z%BaseDashboardLogger.train_epoch_statsc                 C   r0   )NTrainFiguresr#   r&   r   r'   r   r   r&   r   r   r   train_figuresF   r7   z!BaseDashboardLogger.train_figuresc                 C      | j d|||d d S )NTrainAudiosr#   r(   r   r   r)   r   r   r(   r   r   r   r   train_audiosI      z BaseDashboardLogger.train_audiosc                 C   r0   )N	EvalStatsr2   r3   r4   r   r   r   
eval_statsL   r7   zBaseDashboardLogger.eval_statsc                 C   r0   )NEvalFiguresr;   r<   r=   r   r   r   eval_figuresO   r7   z BaseDashboardLogger.eval_figuresc                 C   r?   )N
EvalAudiosrA   rB   rC   r   r   r   eval_audiosR   rE   zBaseDashboardLogger.eval_audiosc                 C   r?   )N
TestAudiosrA   rB   rC   r   r   r   test_audiosU   rE   zBaseDashboardLogger.test_audiosc                 C   r0   )NTestFiguresr;   r<   r=   r   r   r   test_figuresX   r7   z BaseDashboardLogger.test_figuresr   ) __name__
__module____qualname__r   strfloatintr   r   r   r   r   r   r"   r   r%   r'   r)   r+   r,   staticmethodr   r/   r6   r9   r>   rD   rG   rI   rK   rM   rO   r   r   r   r   r	   	   sV    
 

r	   N)abcr   r   typingr   r   
trainer.ior   trainer.utils.distributedr   r	   r   r   r   r   <module>   s
    