o wO i¬ã@sRddlmZddlmmmZddlmZmZde deeeffdd„Z dS)é)ÚDictN)Ú SchedulerÚSchedulerBackendÚ session_nameÚreturncKs&tj|fi|¤Žtj|fi|¤ŽdœS)N)ÚlocalÚdefault)Úlocal_schedulerÚcreate_scheduler)rÚscheduler_params©r úV/home/ubuntu/.local/lib/python3.10/site-packages/torchelastic/tsm/driver/schedulers.pyÚget_schedulerssþr) ÚtypingrÚ'torchelastic.tsm.driver.local_schedulerÚtsmÚdriverr Útorchelastic.tsm.driver.apirrÚstrrr r r r Ús ÿ þ