U vId@sddlZddlZdaedZde_eeejd e dejd dddZ d d Z dd d ZdddZdS)NZanndataFz %(message)sINFOcCs tj|S)z\ Creates a child logger that delegates to anndata_logger instead to logging.root )anndata_loggermanager getLogger)namerH/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/anndata/logging.py get_loggersr cCslddl}|t}z |}Wntk r>|}YnX|dd}|}tdk r`|t}|a||fS)Nri@)psutilProcessosgetpidZ memory_infoAttributeErrorZget_memory_info_previous_memory_usage)r processZmeminfomemZmem_diffrrr get_memory_usages  rcCsF|rdnd}|dkrdnd}|\}}|||d|dd|ddS) N rz ... zMemory usage: current z.2fz GB, difference z+.2fz GBr)Z mem_usagemsgnewlinemorerdiffrrr format_memory_usage's   rcCsttt||dS)N)printrr)rrrrr print_memory_usage1sr)rF)rF)r loggingrrr propagate addHandler StreamHandlerhandlers setFormatter FormattersetLevelr rrrrrrr s