o h@srddlZddlmZddlmZmZddlmZddlm Z ddZ ej dd Z d d Zed d ZddZdS)N)raises) PartialStatenotebook_launcher) require_bnb)is_bnb_availablecCstdtdS)NzPartialState: )printrrra/var/www/html/ai/venv/lib/python3.10/site-packages/accelerate/test_utils/scripts/test_notebook.pybasic_function sr ACCELERATE_NUM_PROCESSEScCsttdtddS)Nr num_processes)rr NUM_PROCESSESrrrr test_can_initializesrcCs@ttddttdtdWddS1swYdS)NzPlease keep these imports)matchrr )rAssertionErrorrr rrrrr test_problematic_importss"rcCs*tdttrtdtdSdS)NzTest basic notebook can be ranzTest problematic imports (bnb))rrrrrrrr mains  r)ospytestr acceleraterraccelerate.test_utilsraccelerate.utilsrr environgetrrrrrrrr s