o wii@sddlZddlZddlmZddlmZmZmZmZm Z ddl m Z er0ddl m Z ddlmZdd Zed eee d ed ffd dZed eee d ffddZdddZed eed ffddZdS)N)metadata) TYPE_CHECKINGListMappingTupleType)logger)Executor)JobEnvironmentcCsdt}t|dr|jddSztdWStyYnwt|dr+|dgSdd|DS)zrReturn an iterable of EntryPoint objects in the 'submitit' group compatible with Python 3.8+ and the backport.selectsubmitit)groupgetcSs g|] }t|dddkr|qS)rNr )getattr).0eprR/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/submitit/core/plugins.py (s z._iter_submitit_entrypoints..)r entry_pointshasattrr TypeErrorr)epsrrr_iter_submitit_entrypointss     rreturnr r c Cs^ddlm}m}ddlm}|j|j|jg}|||g}t D]}|j dvr8t |j d|j q$|j ddd}z|}Wntyd}zt d |d |WYd}~q$d}~ww|j d krp||q$z|} Wn/ty}z#t|d t|d t|} t d| d|d|d |WYd}~q$d}~ww|| q$||fS)Nr)debuglocal)slurm)executorjob_environmentz = :rz Failed to load submitit plugin 'z': rname__name__zFailed to init JobEnvironment 'z' (z) from submitit plugin ')rrr SlurmExecutor LocalExecutor DebugExecutorSlurmJobEnvironmentLocalJobEnvironmentDebugJobEnvironmentrr#r warningvaluesplitload Exception exceptionappendrstr) rrr executorsjob_envs entry_point module_nameobjejob_envr#rrr _get_plugins+s<          r:cCddtdDS)NcSi|]}||qSrr#)rexrrr Uz!get_executors..rr:rrrr get_executorsRsrBcCs|t}dtjvr"tjd}||vsJd|d|d||S|D] }|r0|Sq&tdd|d)N_TEST_CLUSTER_zUnknown $_TEST_CLUSTER_='z', available: .zSCould not figure out which environment the job is runnning in. Known environments: z, )get_job_environmentsosenvironkeysvalues activated RuntimeErrorjoin)envscenvrrrget_job_environmentXs  " rPcCr;)NcSr<rr=)rrOrrrr?lr@z(get_job_environments..r"rArrrrrEjsrE)rr ) functoolsrF importlibrtypingrrrrrcorer core.corer core.job_environmentr r lru_cacher:r2rBrPrErrrrs     $&