
    mi	                         d Z ddlmZ ddlZddlZddlmZmZ ddl	m
Z
 ddlmZ ddlmZ dd	lmZ  eed
      Zde
fdZy)z
Launch command.
    )partialN   )SubmitRulesupdate_from_args)DecoratedMain)Shepherd)
simple_log)reliable_rmtreezLaunch:mainc                    t        |t              }|j                         }t        ||        t	               }t        ||        |j                  | j                        }t        d|        |j                          | j                  rO|j                  t        d       y |j                         rt        d       y |j                  j                          y | j                  rt        d       |j                  |j                  |j                         |j                          t        d       |j                  j                  j!                         rt#        |j                  j                         d |_	        |j%                  |||       |j                          | j&                  s| j(                  rd}d }d}	 	 |j                  j!                         r&|$t+        j,                  d	d
dd|j                  g      }|j                  d      rt        d|j/                                d}nt1        j2                  d       	 |r'|rt1        j2                  d       |j7                          | j(                  rM|sJ|j                  =t        d|j                  j8                          |j                  j                          y y y y y # t4        $ r d}t        d       Y w xY w# |r'|rt1        j2                  d       |j7                          | j(                  rM|sJ|j                  =t        d|j                  j8                          |j                  j                          w w w w xY w)N)logzFetched sheep z!Could not cancel non existing jobzJob is not runningzCanceling current job...zDeleting XP folder...FTtailz-n200z-fforcez"Remote process finished with state   zKeyboardInterrupt received...   z"attach is set, killing remote job )r   r   get_slurm_configr   r   get_sheep_from_argvargvupdatecanceljobis_doneclearcancel_lazycommitxpfolderexistsr
   maybe_submit_lazyr   attachspPopenstatetimesleepKeyboardInterruptkilljob_id)	argsr   shepherdslurmrulessheepdonetail_processwaits	            E/var/www/stems/demucs_env/lib/python3.12/site-packages/dora/launch.pylaunch_actionr3      s   #&H!!#EUD!MEUD!((3E. !OO{{9934
 		 ]]_$% 	 IIzz&'99   +#$88??!!#EHHOO,	ueU3OOyyDKK	'99##%,*>#%88VT5$		,R#SL==)<ekkmLD

2   JJqM!!#{{499(<UYY=M=M<NOPII$$& ) $({-   ! 	1D/0	1 JJqM!!#{{499(<UYY=M=M<NOPII$$& ) $({s,   >BK K$ K!K$  K!!K$ $BM()__doc__	functoolsr   
subprocessr"   r%   confr   r   r   r   shepr   r   r	   utilsr
   r3        r2   <module>r<      s<       /    "j)$9'm 9'r;   