o Tib@sddlZddlZddlZddlZddlmZddlmZddlmZdZ e Z e j dddej ejd d d d Zd dZedkrNe ZeejdSdS)N)date)Path)tabulateiT z--slack_channel_namez trl-push-ci)defaultz)%(asctime)s - %(levelname)s - %(message)s)levelformatc Csg}g}d}zt|q}|D]f}zFt|}|dd}d|vr&|ddnd}|dd} |rU| dkrG|d 7}||||jd dgn||||jd dgWqtjyt} zt d |d | WYd} ~ qd} ~ wwWdn1swYWn9t y} zt d |d| WYd} ~ n!d} ~ wt y} zt d|d | WYd} ~ nd} ~ ww|||fS)Nrnodeiddurationz.4fzN/Aoutcomefailed_zCould not decode line in : z Log file z not found: zError processing log file ) openjsonloadsgetappendstemsplitJSONDecodeErrorloggingwarningFileNotFoundErrorerror Exception) log failed_tests passed_testssection_num_failedflinedata test_namer r er&G/home/ubuntu/.local/lib/python3.10/site-packages/scripts/log_reports.pyprocess_log_file"s@    """ r(c Cspg}d}g}ttd}|stddS|D]F}t|\}}}| o'| } ||7}|| |t|||gzt |Wqt y_} zt d|d| WYd} ~ qd} ~ wwdddtj d d d d d g} |dkr%d } |D]6\} }}|dkr| d | d|d7} dd|D}| dt|ddgddd7} t|r| d| d7} q{td|td| t| tkrd|d} | dd| d d | dddd ddd d d!tj d"d#d$| d%dd&tj d d'td gd(dd)lm}|tj d*d+}|jd,|| | d-dS| ddd.d/d0d td1dS)2Nrz*.logzNo log files found.zCould not remove log file rheader plain_textu🤗 Results of the TEST_TYPEr z TRL tests.)typetext*z failed test(s)* cSsBg|]}|dddd|ddddddgqS)rz::Nz..)r).0testr&r&r' fs6zmain..z ``` z Test Locationz Test Namegrid)headerstablefmtz *z1: Warning! Empty file - check GitHub action job* zTotal failed tests: z### u❌ There are zA failed tests in total! Please check the action results directly.sectionmrkdwnz*For more details:*buttonzCheck Action resultsz0https://github.com/huggingface/trl/actions/runs/ GITHUB_RUN_ID)r,r-url)r,r- accessorycontextz On Push main z results for )r,elements) WebClientSLACK_API_TOKEN)token#)channelr-blocksu/✅ No failures! All tests passed successfully.T)r,r-emojiz%All tests passed. No errors detected.)listrglobrinfor(rstrosremoveOSErrorrenvironrranyprintlenMAX_LEN_MESSAGErtoday slack_sdkr@chat_postMessage)slack_channel_name group_infototal_num_failedtotal_empty_files log_filesrr passedr empty_filer%payloadmessagename num_failedr failed_tabler@ slack_clientr&r&r'mainAs   "      rc__main__)argparserrrKdatetimerpathlibrrrRArgumentParserparser add_argument basicConfigINFOr(rc__name__ parse_argsargsrVr&r&r&r's"   f