mi}ddlmZddlmZmZddlmZddlZddlm Z ddl Z ddl m Z ddlmZdd lmZd e j$e j&d efd Zed GddZGddZeZd efdZd efdZy))contextmanager) dataclassfield)sha1N)Path) DoraConfig)Link)jsonabledeltareturnct|}ttj|j dj ddS)Nutf8)sortedrjsondumpsencode hexdigest)r sorted_deltas A/var/www/stems/demucs_env/lib/python3.12/site-packages/dora/xp.py_get_sigrs:%=L  <(//7 8 B B DRa HHF)initc eZdZUdZeed<ejed<eje ed<e ed<ejejeje ejfed<e dZ eed < ddedejdeje dejejeje ejfdeje f d Zed efd Zed efdZed efdZed efdZed efdZed efdZed efdZed efdZed ejefdZed ejefdZeddefdZy )XPz Represent a single experiment, i.e. a specific set of parameters that is linked to a unique signature. One XP can have multiple runs. doracfgargvsigr F)comparelinkNc B||_||_||_|2t|Dcgc]\}}|j |r||fc}}}||_||Jt |}||_t|j|jjz |_ ycc}}wN) rrrr is_excludedr rr r folderhistoryr")selfrrrr r kvs r__init__z XP.__init__(s   RAd>N>Nq>Qq!fRSE ;$ $$5/Ctyy'8'889 Ss B B r c|jJ|jj|jjz |jz Sr$)r rdirxpsr(s rr&z XP.folder7s6xx###yy}}tyy}},txx77rcb|jjr|jdz StdS)Ncode.)rgit_saver&rr/s r code_folderzXP.code_folder<s( 99  ;;' '9 rc\|j|jjjz Sr$)r&rshepsubmitit_folderr/s r _xp_submititzXP._xp_submititC{{TYY^^;;;;rc\|j|jjjz Sr$)r&rr6latest_submititr/s r_latest_submititzXP._latest_submititGr9rcf|jjr |jS|jSr$)r<existsr8r/s rsubmititz XP.submititKs,  ' ' )(( ($$ $rcH|j|jjz Sr$)r&rrendezvous_filer/s rrAzXP.rendezvous_fileRs{{TYY6666rcH|j|jjz Sr$)r&rr'r/s rr'z XP.historyVs{{TYY....rc |jdz SNz .argv.json)r&r/s r _argv_cachezXP._argv_cacheZs{{\))rc|jj:|jj|jjz |jz Syr$)rsharedr.r r/s r_shared_folderzXP._shared_folder^s; 99   '99##diimm3dhh> >rc:|j|jdz SyrD)rHr/s r_shared_argv_cachezXP._shared_argv_cacheds"    *&&5 5rstackc#jKtj||5ddddy#1swYyxYww)a!Context manager, fake being in the XP for its duration. Set `stack=True` if you want to allow this to happen from within another experiment. ..Warning:: For hydra experiment, this will not convert any path automatically, or setup loggers etc. N)_contextenter_xp)r(rKs renterzXP.enterjs/  tU +     s3' 303)NNF)__name__ __module__ __qualname____doc__r __annotations__tpAnyListstrOptionalTuplerr"r r+propertyrr&r4r8r<r?rAr'rErHrJrboolrOrrrrs   K ''#, H ;;rwwrxxRVV 45 66u%D$%GK)- :Z :bff :BGGCL : BGGBHHS"&&[,A$BC :kk#& :888T **}}R  rc4ttjS)z$Return True if running within an XP.)r]rMrbr^rris_xprls  r) contextlibr dataclassesrrhashlibrrpathlibrtypingrVconfr r"r utilsr rXrWrYrrr`rMrjr]rlr^rrrts&( IBGGBFFOII  [[[|" :!!tr