o 2wi}@sddlmZddlmZmZddlmZddlZddlm Z ddl Z ddl m Z ddlmZdd lmZd e je jd efd d ZeddGdddZGdddZeZd efddZd efddZdS))contextmanager) dataclassfield)sha1N)Path) DoraConfig)Link)jsonabledeltareturncCs(t|}tt|dddS)Nutf8)sortedrjsondumpsencode hexdigest)r sorted_deltarD/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/dora/xp.py_get_sigs rF)initc @seZdZUdZeed<ejed<eje ed<e ed<ej ejej e ejfed<e ddZ eed < d%dedejdeje dej ejej e ejfdej e f d d Zed efddZed efddZed efddZed efddZed efddZed efddZed efddZed efddZed ej efddZed ej efd d!Zed&d"efd#d$Zd S)'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)comparelinkNcsp|_||_||_|durtfdd|D}||_|dur)|dus%Jt|}||_t|j|jj |_ dS)Ncs"g|] \}}|s||fqSr) is_excluded).0kvrrr /s"zXP.__init__..) rrrr r rrr folderhistoryr)selfrrrr rrr$r__init__(s z XP.__init__r cCs$|jdusJ|jj|jj|jSN)rrdirxpsr(rrrr&7sz XP.foldercCs|jjr |jdStdS)Ncode.)rgit_saver&rr-rrr code_folder<s zXP.code_foldercC|j|jjjSr*)r&rshepsubmitit_folderr-rrr _xp_submititCzXP._xp_submititcCr2r*)r&rr3latest_submititr-rrr_latest_submititGr6zXP._latest_submititcCs|jr|jS|jSr*)r8existsr5r-rrrsubmititKs z XP.submititcC|j|jjSr*)r&rrendezvous_filer-rrrr<RzXP.rendezvous_filecCr;r*)r&rr'r-rrrr'Vr=z XP.historycCs |jdSNz .argv.json)r&r-rrr _argv_cacheZ zXP._argv_cachecCs&|jjdur|jj|jj|jSdSr*)rsharedr,rr-rrr_shared_folder^s zXP._shared_foldercCs|jdur |jdSdSr>)rBr-rrr_shared_argv_cacheds  zXP._shared_argv_cachestackccs:t|| dVWddS1swYdS)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(rDrrrenterjs "zXP.enter)NNF)__name__ __module__ __qualname____doc__r__annotations__tpAnyListstrOptionalTuplerrr r)propertyrr&r1r5r8r:r<r'r?rBrCrboolrGrrrrrsJ    rc@s.eZdZd ddZed dedefdd ZdS) _Contextr NcCs g|_dSr*)_xpsr-rrrr){s z_Context.__init__FxprDc csH|jr |s td|j|z dVW|jddS|jdw)NzAlready in a xp.)rW RuntimeErrorappendpop)r(rXrDrrrrF~s  z_Context.enter_xp)r NrH)rIrJrKr)rrrUrFrrrrrVxs rVcCstjstdtjdS)z_When running from within an XP, returns the XP object. Otherwise, raises RuntimeError. z Not in a xp!rY)rErWrZrrrrget_xps r]cCs ttjS)z$Return True if running within an XP.)rUrErWrrrris_xpr@r^) contextlibr dataclassesrrhashlibrrpathlibrtypingrNconfrrr utilsr rPrOrQrrrVrEr]rUr^rrrrs      ^