9i@DddlmZerddlmZddlmZd defdZy) )is_tqdm_available)tqdm) PartialStatemain_process_onlycts tdd}|rtjdk(}t |i|d|iS)z Wrapper around `tqdm.tqdm` that optionally displays only on the main process. Args: main_process_only (`bool`, *optional*): Whether to display the progress bar only on the main process zZAccelerate's `tqdm` module requires `tqdm` to be installed. Please run `pip install tqdm`.Frdisable)r ImportErrorrlocal_process_index_tqdm)rargskwargsr s _/var/www/html/backtest/airagagent/rag_env/lib/python3.12/site-packages/accelerate/utils/tqdm.pyrrsE  vwwG.449 $ 2& 2' 22N)T)importsr tqdm.autorr staterboolrrrs$''  3D 3r