Ë
    @çi7  ã                   ó”   — d dl mZ d dlmZmZ d dlmZ d dlmZm	Z	 ddl
mZ ddlmZ ddlmZ dZd	efd
„Zd	efd„Zd„ Zd	efd„Zd„ Zy)é   )ÚSubmitRules)ÚExplorerÚLauncher)Ú	HydraMain)Úrun_gridÚRunGridArgsé   )Ú	mock_shep)Úget_mainNÚlauncherc                 ó$   —  | «         | d¬«       y )Né*   )Únum_workers© ©r   s    úi/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/dora/tests/test_grid.pyÚ	explore_1r      s   € Ù„JÙ˜Öó    c                 ó   —  | dd¬«       y )Nr   é   )r   Úar   r   s    r   Ú	explore_2r      s   € Ù˜˜qÖ!r   c                 óŠ  ‡‡‡‡— ˆˆˆˆfd„}t        «       5  t        | «      Š‰j                  «       Št        «       Št	        «       Šd‰_        d‰_         |t        «      }t        |«      dk(  sJ ‚|d   j                  �J ‚d‰_         |t        «      }t        |«      dk(  sJ ‚|d   j                  j                  dk(  sJ ‚|d   j                  «       rJ ‚d‰_         |t        «      }t        |«      dk(  sJ ‚|d   j                  «       dk(  sJ ‚|d   j                  «       sJ ‚d‰_         |t        «      }t        |«      dk(  sJ ‚|d   j                  «       dk(  sJ ‚d‰_         |t        «      }t        |«      dk(  sJ ‚|d   j                  «       dk(  sJ ‚|d   j                  j                  d	k(  sJ ‚|d   }d‰_         |t         «      }t        |«      dk(  sJ ‚|d   j                  «       dk(  sJ ‚|d   j                  j                  d
k(  sJ ‚|j                  «       dk(  sJ ‚	 d d d «       y # 1 sw Y   y xY w)Nc                 ó8   •— t        ‰t        | «      d‰‰‰¬«      S )NÚunittest)ÚslurmÚrulesÚargs©r   r   )Úexplorer   Úmainr   r   s    €€€€r   Úrgridztest_shep.<locals>.rgrid   s"   ø€ Ü˜œh wÓ/°Ø#¨5°tô=ð 	=r   FTr	   é    Ú0Ú	CANCELLEDÚUNKNOWNÚ1Ú2)r
   r   Úget_slurm_configr   r   ÚmonitorÚdry_runr   ÚlenÚjobÚjob_idÚis_doneÚcancelÚstateÚretryÚverboser   )Útmpdirr"   ÚsheepsÚ	old_sheepr   r!   r   r   s       @@@@r   Ú	test_shepr7      s/  û€ ÷=ô 
�Ü˜ÓˆØ×%Ñ%Ó'ˆÜ“ˆÜ‹}ˆàˆŒØˆŒá”yÓ!ˆÜ�6‹{˜aÒÐÐØ�a‰y�}‰}Ð$Ð$Ð$àˆŒÙ”yÓ!ˆÜ�6‹{˜aÒÐÐØ�a‰y�}‰}×#Ñ# sÒ*Ð*Ð*Ø˜!‘9×$Ñ$Ô&Ð&Ð&àˆŒÙ”yÓ!ˆÜ�6‹{˜aÒÐÐØ�a‰y�‰Ó  KÒ/Ð/Ð/Ø�a‰y× Ñ Ô"Ð"Ð"àˆŒÙ”yÓ!ˆÜ�6‹{˜aÒÐÐØ�a‰y�‰Ó  KÒ/Ð/Ð/àˆŒÙ”yÓ!ˆÜ�6‹{˜aÒÐÐØ�a‰y�‰Ó  IÒ-Ð-Ð-Ø�a‰y�}‰}×#Ñ# sÒ*Ð*Ð*à˜1‘Iˆ	àˆŒÙ”yÓ!ˆÜ�6‹{˜aÒÐÐØ�a‰y�‰Ó  IÒ-Ð-Ð-Ø�a‰y�}‰}×#Ñ# sÒ*Ð*Ð*Ø�‰Ó  KÒ/Ð/Ñ/÷W 
�‰ús   —HH9È9Ic                 ó~   — | j                  ddd dœ«        | dddigi«        | ddd	d
œi«        | ddd	ddœi«       y )Né2   Ú123)Úepochsz
optim.lossr   z	complex.aÚtestÚweirdz	complex.bé   r   ©r   Úbz
+complex.bé   ©r   r@   Úc)Úbind_r   s    r   Úexplore_hydrarE   M   sP   € Ø‡N�N˜b°ÀdÑKÔLÙˆk˜V WÐ-Ð.Ð/Ô0Ùˆk ¨!Ñ,Ð-Ô.Ùˆl "¨1°2Ñ6Ð7Õ8r   c                 óÀ  ‡‡‡— ˆˆˆfd„}dt         _        t        «       5  t        | «      Št	        «       Št        «       Šd‰_        d‰_         |t        «      }t        |«      dk(  sJ ‚|d   j                  j                  }|j                  dk(  sJ ‚|j                  j                  dk(  sJ ‚|j                  �J ‚|j                   j"                  dd	igk(  sJ ‚|d
   j                  j                  }|j                   j$                  dddœk(  sJ ‚|d   j                  j                  }|j                   j$                  ddddœk(  sJ ‚	 d d d «       y # 1 sw Y   y xY w)Nc                 ó6   •— t        ‰t        | «      d‰‰¬«      S )Nr   )r   r   r   )r    r   r!   r   s    €€€r   r"   ztest_shep_hydra.<locals>.rgridU   s    ø€ Ü˜œh wÓ/°Ø#¨$ô0ð 	0r   FTé   r#   r9   r:   r<   r=   r	   r>   r   r?   r   rA   rB   )r   Ú_slowr
   Úget_main_hydrar   r   r*   r+   rE   r,   ÚxpÚcfgr;   ÚoptimÚlossr   Úcomplexr   r@   )r4   r"   r5   rL   r   r!   r   s       @@@r   Útest_shep_hydrarP   T   s)  ú€ ö0ð „I„OÜ	�Ü˜fÓ%ˆÜ“ˆÜ‹}ˆØˆŒØˆŒá”}Ó%ˆÜ�6‹{˜aÒÐÐØ�Q‰i�l‰l×ÑˆØ�z‰z˜RÒÐÐØ�y‰y�~‰~ Ò&Ð&Ð&Ø�‰Ð&Ð&Ð&Ø�{‰{�}‰} &¨'Ð!2Ð 3Ò3Ð3Ð3à�Q‰i�l‰l×ÑˆØ�{‰{�}‰} b¨qÑ 1Ò1Ð1Ð1à�Q‰i�l‰l×ÑˆØ�{‰{�}‰} b¨q°rÑ :Ò:Ð:Ñ:÷' 
�‰ús    D*EÅE)Úconfr   r    r   r   Úhydrar   Úgridr   r   Ú	fake_shepr
   Ú	test_mainr   Ú
test_hydrarJ   Ú_retr   r   r7   rE   rP   r   r   r   Ú<module>rX      sP   ðõ ß (Ý ß (Ý  Ý Ý 2à€ð˜ó ð
"˜ó "ò/0ðd9˜Hó 9ó;r   