o
    XεiM                     @   sN   d dl Z d dlZd dlmZ d dlmZ e dZdd Zdd Z	d	d
 Z
dS )    N)ConsoleLogger)DummyLoggertrainerc                   C      dt jv r
t jd S d S )NMLFLOW_TRACKING_URIosenviron r
   r
   L/home/ubuntu/.local/lib/python3.10/site-packages/trainer/logging/__init__.pyget_mlflow_tracking_url      

r   c                   C   r   )NAIM_TRACKING_URIr   r
   r
   r
   r   get_ai_repo_url   r   r   c                 C   s  | j }| j}| jr| jn|}| jdkr2ddlm} |r"| d| n|}|||d}td| |S | jdkrHddlm	} |||| | j
d	}|S | jd
kr[ddlm}	 |	||d}|S | jdkrnddlm}
 |
||d}|S | jdkrddlm} |||||d}|S td| j )Ntensorboardr   )TensorboardLogger@)
model_namez- > Start Tensorboard: tensorboard --logdir=%swandb)WandbLogger)projectnameconfigentitymlflow)MLFlowLogger)log_urir   aim)	AimLogger)repor   clearml)ClearMLLogger)
output_uri
local_pathproject_name	task_namezUnknown dashboard logger: )run_namer$   
logger_uridashboard_logger"trainer.logging.tensorboard_loggerr   loggerinfotrainer.logging.wandb_loggerr   wandb_entitytrainer.logging.mlflow_loggerr   trainer.logging.aim_loggerr   trainer.logging.clearml_loggerr!   
ValueError)r   output_pathr&   r$   r   r   r   r(   r   r   r   r!   r
   r
   r   logger_factory   sB   
 



r3   )loggingr   trainer.logging.console_loggerr   trainer.logging.dummy_loggerr   	getLoggerr*   r   r   r3   r
   r
   r
   r   <module>   s    
