o h @sUddlZddlZddlZddlmZmZmZddlmZddl m Z ddl m m Z ddlmZeZgZeeed<e eaedZedZd eeeffd d Zd eeeffd d Z ddeded eeeefgeeefffddZdefddZ dS)N)AnyCallableTypeVar) ParamSpec)uuid4)DCP_LOGGER_NAME__all___T_PreturncOsi}|dd}|dd}|dd}|dd}|s(|p|}r(t|dd}|dur0t|n||d<ttj|d<|rD|jj|d<|rL|jj|d<|rT|jj|d<|S)z0 Extracts log data from dcp method args storage_writerNstorage_readerplanner checkpoint_iduuid)getgetattrstrrint __class____name__)argskwargsmsg_dictr r rr serializerrw/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/distributed/checkpoint/logger.py_msg_dict_from_dcp_method_argss"        rcOs0t|i|}|tj|g|Ri||S)N)rupdate c10d_logger _get_msg_dict) func_namerrrrrrr ;sr Flog_exceptionswrapper_kwargsc s dtttfffdd }|S)zKThis method decorator logs the start, end, and exception of wrapped events.funccs0tdtjdtjdtffdd }|S)Nrrr c stjg|Rii|}d|d<t}||d<|d<t|z |i|}Wn#tyQ}zrLd|d<||d<t|d<t|d}~wwd|d<t}t|d<|||d<t||S) Nstarteventtimer" exceptionerrorend times_spent)r rr'time_ns _dcp_loggerdebug BaseExceptionr))rrrt0resultr)t1)r$r"r#rrwrapperHs:         z6_dcp_method_logger..decorator..wrapper) functoolswrapsr rrr )r$r3r"r#)r$r decoratorGs$ z%_dcp_method_logger..decorator)rr r )r"r#r7rr6r_dcp_method_loggerBs$r8rankcCsJttjt}|tjtd|d}||t|dS)N[z6] %(asctime)s - %(name)s - %(levelname)s - %(message)s)loggersetLevelloggingINFO StreamHandler Formatter setFormatter addHandler)r9ch formatterrrr _init_loggerns    rE)F)!r4r=r'typingrrrtyping_extensionsrrrtorch.distributed.c10d_logger distributedr-torch.distributed.checkpoint.logging_handlersr getLoggerr;rlistr__annotations___get_or_create_loggerr-r r dictrr boolr8rrErrrrs0     " ,