Ë
    @çiÏ  ã                   ó¨   — d dl mZ d dlZd dlZddlmZ ddlmZmZm	Z	m
Z
 ddlmZmZ da ed«      j                  «       Zd„ Zd	„ Zd
„ Zd„ Zd„ Zd„ Zy)é    )ÚPathNé   )Ú
hydra_main)Úassign_cloneÚget_new_cloneÚenter_cloneÚto_absolute_path)Úget_xpÚXPÚ.c                 ó¸   — t        «       }|j                  j                  ddi«       |at	        d«      t        t        «      k(  sJ t	        d«      t        f«       ‚y )NÚlossçš™™™™™¹?r   )r
   ÚlinkÚpush_metricsÚ_retr	   ÚstrÚcurrent_path)ÚcfgÚxps     új/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/dora/tests/test_hydra.pyÚ_mainr      sL   € ä	‹€BØ‡G�G×Ñ˜& #˜Ô'Ø€DÜ˜CÓ ¤C¬Ó$5Ò5Ð\Ô8HÈÓ8MÌ|Ð7\Ó\Ñ5ó    c                 ó~   — t        t        | «      «      }  t        dd¬«      t        «      }| |j                  _        |S )Nz./test_confÚ	test_conf)Úconfig_pathÚconfig_name)r   r   r   r   ÚdoraÚdir)ÚtmpdirÚ	dora_mains     r   Úget_mainr"      s5   € Ü”#�f“+Ó€FØN”
 }À+ÔNÌuÓU€IØ€I‡N�NÔØÐr   c                 ó¬   — t        t        j                  «      }	 |t        j                  dd   | «        |t        _        t        S # |t        _        w xY w)Né   )ÚlistÚsysÚargvr   )Úmainr'   Úold_argvs      r   Úcallr*   $   s?   € Ü”C—H‘H‹~€HðØŒ�‰��ˆÙŒàŒŒÜ€Køð Œ�ús   ›A ÁAc                 ó*  — t         t        _        t        | «      }dg}|j	                  |«      }|j                  |«       d|j                  _        t        |«      }t        ||«       t        |«      5  t        ||«       d d d «       y # 1 sw Y   y xY w)Nzoptim.loss=git_saveT)Ú__name__r   Ú
__module__r"   r
   Úinit_xpr   Úgit_saver   r   r   r*   )r    r(   r'   r   Úclones        r   Útest_hydra_git_saver1   .   st   € Ü„EÔÜ�FÓ€DØ!Ð"€DØ	�‰�TÓ	€BØ‡L�L�ÔØ€B‡G�GÔä˜$Ó€EÜ��UÔÜ	�UÕ	ÜˆT�4Ô÷ 
×	Ñ	ús   Á3B	Â	Bc                 ó  — t         t        _        t        | «      }t	        |g «      }t        |t        «      sJ ‚t        |j                  «      dkD  sJ ‚|j                  «       j                  dk(  sJ ‚dg}t	        ||«      }|j                  |j                  k(  sJ ‚|j                  j                  dk(  sJ ‚|j                  ddi«      }t        |«      dkD  sJ ‚t	        ||«      }|j                  |j                  k(  sJ ‚|j                  j                  j                  dk(  sJ ‚ddg}|j                  |«      }||k(  sJ ‚t	        ||«      }|j                  |j                  k7  sJ ‚||j!                  |j                  «      k(  sJ ‚|j#                  |j                  «      }|j$                  |j$                  k(  sJ ‚|j&                  |j&                  k(  sJ ‚|j                  |j                  k(  sJ ‚|j(                  |j(                  k(  sJ ‚|j+                  |«      }|d	   d
   dk(  sJ ‚|j-                  |«      }|dk(  sJ ‚dg}t	        ||«      }|j                  j.                  dk(  sJ ‚t1        j2                  t4        «      5  |j                  d«       d d d «       dg}t	        ||«      }|j                  j6                  j8                  dk(  sJ ‚t;        |j                  d«      rJ ‚dg}t	        ||«      }|j                  j<                  j                  dk(  sJ ‚t;        |j                  d«      rJ ‚ddg}t1        j2                  t>        «      5  t	        ||«      }d d d «       y # 1 sw Y   ŒÔxY w# 1 sw Y   y xY w)Nr   é   znum_workers=40é(   z	useless.aé   zuseless.b=falsezoptim.loss=l1éÿÿÿÿr   r   zopt.loss=l1z+k=youpiÚyoupig      à?zplop.b=5Úlapinzgroup=lapinÚplop) r,   r   r-   r"   r*   Ú
isinstancer   ÚlenÚsigÚget_slurm_configÚcpus_per_taskr   Únum_workersÚvalue_to_argvÚuselessÚaÚget_argv_from_sigÚget_xp_from_sigr'   Údeltar   Úget_xp_historyÚget_nameÚkÚpytestÚraisesÚ
ValueErrorr9   ÚbÚhasattrr8   Ú	Exception)	r    r(   r   r'   Úxp2ÚpreÚxp3ÚmetricsÚnames	            r   Ú
test_hydrarT   <   s   € Ü„EÔÜ�FÓ€DÜ	ˆd�B‹€BÜ�bœ"ÔÐÐÜˆr�v‰v‹;˜Š?Ðˆ?à× Ñ Ó"×0Ñ0°AÒ5Ð5Ð5àÐ€DÜ
ˆt�TÓ
€CØ�6‰6�S—W‘WÒÐÐØ�7‰7×Ñ "Ò$Ð$Ð$à×Ñ˜{¨AÐ.Ó/€DÜˆt‹9�qŠ=Ðˆ=Ü
ˆt�TÓ
€CØ�6‰6�S—W‘WÒÐÐØ�7‰7�?‰?×Ñ Ò!Ð!Ð!à˜oÐ
.€CØ×Ñ˜cÓ"€DØ�3Š;Ðˆ;Ü
ˆt�S‹/€CØ�6‰6�S—W‘WÒÐÐà�4×)Ñ)¨#¯'©'Ó2Ò2Ð2Ð2à
×
Ñ
˜sŸw™wÓ
'€CØ�8‰8�s—x‘xÒÐÐØ�9‰9˜Ÿ	™	Ò!Ð!Ð!Ø�7‰7�c—g‘gÒÐÐØ�8‰8�s—x‘xÒÐÐà×!Ñ! #Ó&€GØ�2‰;�vÑ #Ò%Ð%Ð%à�=‰=˜Ó€DØ�=Ò Ð Ð àˆ<€DÜ
ˆt�TÓ
€CØ�7‰7�9‰9˜ÒÐÐä	�‰”zÕ	"Ø×Ñ˜3Ô÷ 
#ð ˆ<€DÜ
ˆt�TÓ
€CØ�7‰7�<‰<�>‰>˜QÒÐÐÜ�s—w‘w Ô(Ð(Ð(àˆ?€DÜ
ˆt�TÓ
€CØ�7‰7�=‰=�?‰?˜aÒÐÐÜ�s—w‘w Ô'Ð'Ð'à˜:Ð&€DÜ	�‰”yÕ	!Ü�4˜Óˆ÷ 
"Ð	!÷ 
#Ð	"ú÷ 
"Ð	!ús   Ê	M.ÍM:Í.M7Í:Nc                 óª  — t         t        _        t        | «      }t	        |g «      }t        |j                  j                  «       |j                  j                  j                  g d¢k(  sJ ‚t	        |dg«      }|j                  j                  j                  dgk(  sJ ‚t	        |dg«      }|j                  j                  j                  dddœk(  sJ ‚t	        |dg«      }|j                  j                  j                  d	ddœk(  sJ ‚|j                  d
d	ddœi«      }t	        ||«      }|j                  j                  j                  d	ddœk(  sJ ‚y )N)r$   r   r5   zcomplex.a=[0]r   zcomplex.b.a=50é2   r   )rB   rL   zcomplex.b={a:21}é   z	complex.bé4   )r,   r   r-   r"   r*   Úprintr   ÚcomplexrB   rL   r@   )r    r(   r   r'   s       r   Útest_complex_typesr[   z   s!  € ä„EÔÜ�FÓ€DÜ	ˆd�B‹€BÜ	ˆ"�&‰&�.‰.ÔØ�6‰6�>‰>×ÑšyÒ(Ð(Ð(Ü	ˆd�_Ð%Ó	&€BØ�6‰6�>‰>×Ñ ˜sÒ"Ð"Ð"Ü	ˆdÐ%Ð&Ó	'€BØ�6‰6�>‰>×Ñ R¨aÑ0Ò0Ð0Ð0Ü	ˆdÐ'Ð(Ó	)€BØ�6‰6�>‰>×Ñ R¨aÑ0Ò0Ð0Ð0Ø×Ñ˜{°"¸2Ñ,>Ð?Ó@€DÜ	ˆd�DÓ	€BØ�6‰6�>‰>×Ñ R¨bÑ1Ò1Ð1Ñ1r   )Úpathlibr   r&   rI   Úhydrar   r/   r   r   r   r	   r   r
   r   r   Úresolver   r   r"   r*   r1   rT   r[   © r   r   Ú<module>r`      sS   ðõ Û 
ã å ß QÓ Qß à€á�C‹y× Ñ Ó"€ò]òòòò;ó|2r   