
    eThb                     X   S SK Jr  SSKJrJrJrJrJrJrJ	r	J
r
  SSKJrJrJrJrJr  SSKJr  \R&                  " \5      r " S S5      r " S S	\5      r " S
 S\5      r " S S\5      r " S S\5      r\\\\4 V s0 s H  n \" U R6                  5      U _M     sn rS\4S jrgs  sn f )    )Optional   )is_optuna_availableis_ray_tune_availableis_sigopt_availableis_wandb_availablerun_hp_search_optunarun_hp_search_rayrun_hp_search_sigoptrun_hp_search_wandb)HPSearchBackenddefault_hp_space_optunadefault_hp_space_raydefault_hp_space_sigoptdefault_hp_space_wandb)loggingc                   r    \ rS rSr% \\S'   Sr\\   \S'   \S 5       r	S\
S\4S jrS	 rS
 r\S 5       rSrg)HyperParamSearchBackendBase'   nameNpip_packagec                      [         eNNotImplementedError     Z/var/www/auris/envauris/lib/python3.13/site-packages/transformers/hyperparameter_search.pyis_available(HyperParamSearchBackendBase.is_available+   s    !!r   n_trials	directionc                     [         er   r   selftrainerr!   r"   kwargss        r   runHyperParamSearchBackendBase.run/       !!r   c                     [         er   r   r%   trials     r   default_hp_space,HyperParamSearchBackendBase.default_hp_space2   r*   r   c                     U R                  5       (       d*  [        SU R                   SU R                  5        S35      eg )NzYou picked the z' backend, but it is not installed. Run .)r   RuntimeErrorr   pip_install)r%   s    r   ensure_available,HyperParamSearchBackendBase.ensure_available5   sF      ""!$)),STXTdTdTfSgghi  #r   c                 H    SU R                   =(       d    U R                   S3$ )Nz`pip install `)r   r   )clss    r   r3   'HyperParamSearchBackendBase.pip_install;   s    s:#((;1==r   r   )__name__
__module____qualname____firstlineno__str__annotations__r   r   staticmethodr   intr(   r.   r4   classmethodr3   __static_attributes__r   r   r   r   r   '   sX    
I!%K#%" ""S "S "" > >r   r   c                   @    \ rS rSrSr\S 5       rS\S\4S jr	S r
Srg	)
OptunaBackend@   optunac                      [        5       $ r   )r   r   r   r   r   OptunaBackend.is_availableC       "$$r   r!   r"   c                     [        XU40 UD6$ r   )r	   r$   s        r   r(   OptunaBackend.runG       #GyKFKKr   c                     [        U5      $ r   )r   r,   s     r   r.   OptunaBackend.default_hp_spaceJ       &u--r   r   Nr:   r;   r<   r=   r   r@   r   rA   r>   r(   r.   rC   r   r   r   rE   rE   @   4    D% %LS LS L.r   rE   c                   D    \ rS rSrSrSr\S 5       rS\S\	4S jr
S rS	rg
)RayTuneBackendN   rayz'ray[tune]'c                      [        5       $ r   )r   r   r   r   r   RayTuneBackend.is_availableR   s    $&&r   r!   r"   c                     [        XU40 UD6$ r   )r
   r$   s        r   r(   RayTuneBackend.runV   s     IHHHr   c                     [        U5      $ r   )r   r,   s     r   r.   RayTuneBackend.default_hp_spaceY   s    #E**r   r   N)r:   r;   r<   r=   r   r   r@   r   rA   r>   r(   r.   rC   r   r   r   rT   rT   N   s9    DK' 'IS IS I+r   rT   c                   @    \ rS rSrSr\S 5       rS\S\4S jr	S r
Srg	)
SigOptBackend]   sigoptc                      [        5       $ r   )r   r   r   r   r   SigOptBackend.is_available`   rJ   r   r!   r"   c                     [        XU40 UD6$ r   )r   r$   s        r   r(   SigOptBackend.rund   rM   r   c                     [        U5      $ r   )r   r,   s     r   r.   SigOptBackend.default_hp_spaceg   rP   r   r   NrQ   r   r   r   r^   r^   ]   rR   r   r^   c                   @    \ rS rSrSr\S 5       rS\S\4S jr	S r
Srg	)
WandbBackendk   wandbc                      [        5       $ r   )r   r   r   r   r   WandbBackend.is_availablen   s    !##r   r!   r"   c                     [        XU40 UD6$ r   )r   r$   s        r   r(   WandbBackend.runr   s    "7iJ6JJr   c                     [        U5      $ r   )r   r,   s     r   r.   WandbBackend.default_hp_spaceu   s    %e,,r   r   NrQ   r   r   r   rh   rh   k   s4    D$ $KS KS K-r   rh   returnc                     [         R                  5        V s/ s H  o R                  5       (       d  M  U PM     nn [        U5      S:  aD  US   R                  n[        U5      S:  a$  [
        R                  [        U5       SU S35        U$ [        SSR                  S [         R                  5        5       5      -   5      es  sn f )Nr   r   z1 hyperparameter search backends available. Using z as the default.z,No hyperparameter search backend available.

c              3   b   #    U  H%  nS UR                    SUR                  5        3v   M'     g7f)z - To install z run N)r   r3   ).0backends     r   	<genexpr>,default_hp_search_backend.<locals>.<genexpr>   s2      
F W\\N%0C0C0E/FGFs   -/)	"ALL_HYPERPARAMETER_SEARCH_BACKENDSvaluesr   lenr   loggerinfor2   join)rv   available_backendsr   s      r   default_hp_search_backendr   ~   s    1S1Z1Z1\w1\g`t`t`v'1\w
"!!$))!"Q&KK)*++\]a\bbrs 
7
)) 
=DDF
 
	
  xs
   CCN)typingr   integrationsr   r   r   r   r	   r
   r   r   trainer_utilsr   r   r   r   r   utilsr   
get_loggerr:   r|   r   rE   rT   r^   rh   r   ry   r>   r   )rv   s   0r   <module>r      s    	 	 	   
		H	%> >2./ .+0 +./ .-. - <I.Zgiu:v&:vwOGLL!7*:v& "
3 &s   ?B'