ukic UddlZddlZddlmZejddZiZeee fe d<ejejZ e jeejej ej"ej$ej$ej&ej(ej*dZddd d d ddd Zdd edzfd Zd edzddfdZej4dZGddej8ZejejZej?ej(ejeejAegZ!dZ"dZ#dedzfdZ$y)N)utilsz0{levelname}:{asctime}:{name}:{lineno}: {message}{)style_logging_level_set)CRITICALFATALERRORWARNWARNINGINFODEBUGNOTSET)rrr r r r r logging_levelc |dvry|tvr-td|dttjdt|}t j |y)N)NrzAttempting to set log level "z$" which isn't one of the supported: .) _tf_cpp_map ValueErrorlistkeysrabsl_set_min_log_level)r log_levels R/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/_src/logging_config.py_set_cpp_min_log_levelr2se&& +% 5m_E{//12316 77-()y)returnctjD]@\}}tj|}|j t |j |Btjt||yt|}tjdtjdg}|D]E}|j ||jt |jt|j<Gy)Njaxjaxlib) ritemslogging getLogger removeHandler_jax_logger_handlersetLevelclearr _nameToLevel addHandlerlevelname)r logger_namer+loggerlogging_level_num root_loggerss rupdate_logging_level_globalr1>s.446k5   { +F ,- OOE' "=1##E*G,=,=h,GH,3f OO%& )*&,llv{{#3rr ceZdZdZy)_DebugHandlerFiltercDtjtjkDSN) _jax_loggerr+r#r )self_s rfilterz_DebugHandlerFilter.filterXs   w}} ,,rN)__name__ __module__ __qualname__r9rrr3r3Ws-rr3ctj|}tj||jf|j t |jtjy)zMakes the specified logger log everything to stderr. Also adds more useful debug information to the log messages, e.g. the time. Args: logger_name: the name of the logger, e.g. "jax._src.xla_bridge". N) r#r$_debug_enabled_loggersappendr+r*_debug_handlerr'r )r-r.s r_enable_debug_loggingrBbsI   [ )& 67N#//'-- rctD]+\}}|jt|j|-tj y)zDisables all debug logging enabled via `enable_debug_logging`. The default logging behavior will still be in effect, i.e. WARNING and above will be logged to stderr without extra message formatting. N)r?r%rAr'r()r. prev_levels r_disable_all_debug_loggingrEqs@ 3 fj ( OOJ  rmodule_names_strcdt|sy|jd}|D] }t|y)N,)rEsplitrB)rF module_names module_names rupdate_debug_log_modulesrL}s5  !'',,!'k+&'rr5)%r#sys jax._src.libr Formatterlogging_formatterrdictstrint__annotations__ StreamHandlerstderrr& setFormatterrrr r r r rr)rrr1r$r6Filterr3rAr' addFilterr?rBrErLr=rrrZs} &G%%6cC&(DcN'+g++CJJ7 !23  ]] ]] OO LL ]]nn        *#* *3sTz3d3. g& -'..-'&&szz2 &-.,./ ! !'sTz'r