o 2wi@s~dZddlZddlmZddlZddlZddlmZddl m Z ddl m Z de fd d Zddeje d eje fd d ZdS)z=Internal utilities, likely shouldn't be called from outside. N)Path)fatal) DecoratedMain)import_or_fatal main_modulecCstd}g}|D]}|r%|dr%||dr%||jq t|dkr2tddSt|dkr<|dStdd|d dS) N.z __init__.pyz.pyrzCould not find a training package. Use -P, or set DORA_PACKAGE to set the package. Use --main_module or set DORA_MAIN_MODULE to set the module to be excecuted inside the defined package.rzFound multiple candidates: z, z. Use -P, or set DORA_PACKAGE to set package being searched. Use --main_module or set DORA_MAIN_MODULE to set the module being searched inside the package.) riterdiris_direxistsappendnamelenrjoin)rcwd candidateschildrH/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/dora/_utils.py _find_packages     rpackagecCs|dur tjdp d}|durtjd}|durt|}|d|}tjdttd t |}z|j }Wnt yKt d|dYnwt|tsXt |d|S)NDORA_MAIN_MODULEtrain DORA_PACKAGErrz"Could not find function `main` in z).main was not decorated with `dora.main`.)osenvirongetrsyspathinsertstrrresolvermainAttributeErrorr isinstancer)rr module_namemoduler"rrrget_main's"     r')NN)__doc__rpathlibrrtypingtplogrr"rutilsrr rOptionalr'rrrrs    $