o }o™iÈ ã@s<ddlZddlmZddlmZiZdZdd„Z dd„Z dS)éN)Ú MemoryHandlerFcCsVts)tdƒtd<td dd„¡tdƒtd<tj td¡tj td¡dSdS)a  Adds two MemoryHandlers to pytorch_lightning's logger. These two handlers are essentially message buffers. This function is called in nemo.utils.__init__.py. These handlers are used in add_filehandlers_to_pl_logger to flush buffered messages to files. éÿÿÿÿÚ memory_errcSó |jtjkS©N©ÚlevelnoÚ_loggingÚINFO©Úrecord©r úU/home/ubuntu/.local/lib/python3.10/site-packages/nemo/utils/lightning_logger_patch.pyÚ ó z2add_memory_handlers_to_pl_logger..Ú memory_allN)ÚHANDLERSrÚ addFilterÚplÚ_loggerÚ addHandlerr r r rÚ add_memory_handlers_to_pl_loggers  ûrcCs¶t |¡td<tj td¡t |¡td<td dd„¡tj td¡t dd¡r?td td¡td  ¡td=t dd¡rYtd td¡td  ¡td=dSdS)aŒ Adds two filehandlers to pytorch_lightning's logger. Called in nemo.utils.exp_manager(). The first filehandler logs all messages to all_log_file while the second filehandler logs all WARNING and higher messages to err_log_file. If "memory_err" and "memory_all" exist in HANDLERS, then those buffers are flushed to err_log_file and all_log_file respectively, and then closed. ÚfileÚfile_errcSrrrr r r rr0rz/add_filehandlers_to_pl_logger..rNr) r Ú FileHandlerrrrrrÚgetÚ setTargetÚclose)Ú all_log_fileÚ err_log_filer r rÚadd_filehandlers_to_pl_logger&s     ýr ) Úloggingr Úlogging.handlersrÚlightning.pytorchÚpytorchrrÚPATCHEDrr r r r rÚs