o bi>@sddlZddlZddlZddlmZedkrwejddZedede \Z Z ej Z e ZedkrVe jZe ejdd<ejdeZejeZejedSdd lmZee jd d\ZZe eZ!e"e!eZ#e#dSdS) N)MPI__main__zSetup MPI worker) description worker_entry main_entry)mpi_init.)$argparse importlibsysmpi4pyr__name__ArgumentParserparser add_argumentparse_known_argsargsremaining_args COMM_WORLDcommGet_rankrankr entry_fileargvutilspec_from_file_locationspecmodule_from_specmodloader exec_moduleray.runtime_envrrrsplitmodulefunc import_modulemgetattrfr*r*W/home/ubuntu/.local/lib/python3.10/site-packages/ray/_private/runtime_env/mpi_runner.pys0