Ë
    S^(hb  ã                   óP  — d dl mZ ddlmZmZmZmZmZmZm	Z	m
Z
 ddlmZmZmZmZmZ ddlmZ  ej&                  e«      Z G d„ d«      Z G d„ d	e«      Z G d
„ de«      Z G d„ de«      Z G d„ de«      ZeeeefD � ci c]  }  e| j6                  «      | “Œ c} Zdefd„Zyc c} w )é    )Ú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                   óh   — e Zd ZU eed<   dZee   ed<   ed„ «       Zde	defd„Z
d„ Zd	„ Zed
„ «       Zy)ÚHyperParamSearchBackendBaseÚnameNÚpip_packagec                  ó   — t         ‚©N©ÚNotImplementedError© ó    ú`/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/transformers/hyperparameter_search.pyÚis_availablez(HyperParamSearchBackendBase.is_available+   s   € ä!Ð!r   Ún_trialsÚ	directionc                 ó   — t         ‚r   r   ©ÚselfÚtrainerr   r    Úkwargss        r   ÚrunzHyperParamSearchBackendBase.run/   ó   € Ü!Ð!r   c                 ó   — t         ‚r   r   ©r#   Útrials     r   Údefault_hp_spacez,HyperParamSearchBackendBase.default_hp_space2   r'   r   c                 óx   — | j                  «       s*t        d| j                  › d| j                  «       › d�«      ‚y )NzYou picked the z' backend, but it is not installed. Run ú.)r   ÚRuntimeErrorr   Úpip_install)r#   s    r   Úensure_availablez,HyperParamSearchBackendBase.ensure_available5   sC   € Ø× Ñ Ô"ÜØ! $§)¡) Ð,SÐTX×TdÑTdÓTfÐSgÐghÐióð ð #r   c                 ó>   — d| j                   xs | j                  › d�S )Nz`pip install ú`)r   r   )Úclss    r   r/   z'HyperParamSearchBackendBase.pip_install;   s   € à˜sŸ™Ò:°#·(±(Ð;¸1Ð=Ð=r   )Ú__name__Ú
__module__Ú__qualname__ÚstrÚ__annotations__r   r   Ústaticmethodr   Úintr&   r+   r0   Úclassmethodr/   r   r   r   r   r   '   sX   … Ø
ƒIØ!%€K�˜#‘Ó%àñ"ó ð"ð" Sð "°Só "ò"òð ñ>ó ñ>r   r   c                   ó6   — e Zd ZdZed„ «       Zdedefd„Zd„ Z	y)ÚOptunaBackendÚoptunac                  ó   — t        «       S r   )r   r   r   r   r   zOptunaBackend.is_availableC   ó   € ä"Ó$Ð$r   r   r    c                 ó   — t        |||fi |¤ŽS r   )r	   r"   s        r   r&   zOptunaBackend.runG   ó   € Ü# G¨X°yÑKÀFÑKÐKr   c                 ó   — t        |«      S r   )r   r)   s     r   r+   zOptunaBackend.default_hp_spaceJ   ó   € Ü& uÓ-Ð-r   N©
r4   r5   r6   r   r9   r   r:   r7   r&   r+   r   r   r   r=   r=   @   ó4   „ Ø€Dàñ%ó ð%ðL Sð L°Só Ló.r   r=   c                   ó:   — e Zd ZdZdZed„ «       Zdedefd„Z	d„ Z
y)	ÚRayTuneBackendÚrayz'ray[tune]'c                  ó   — t        «       S r   )r   r   r   r   r   zRayTuneBackend.is_availableR   s   € ä$Ó&Ð&r   r   r    c                 ó   — t        |||fi |¤ŽS r   )r
   r"   s        r   r&   zRayTuneBackend.runV   s   € Ü  ¨(°IÑHÀÑHÐHr   c                 ó   — t        |«      S r   )r   r)   s     r   r+   zRayTuneBackend.default_hp_spaceY   s   € Ü# EÓ*Ð*r   N)r4   r5   r6   r   r   r9   r   r:   r7   r&   r+   r   r   r   rH   rH   N   s9   „ Ø€DØ€Kàñ'ó ð'ðI Sð I°Só Ió+r   rH   c                   ó6   — e Zd ZdZed„ «       Zdedefd„Zd„ Z	y)ÚSigOptBackendÚsigoptc                  ó   — t        «       S r   )r   r   r   r   r   zSigOptBackend.is_available`   r@   r   r   r    c                 ó   — t        |||fi |¤ŽS r   )r   r"   s        r   r&   zSigOptBackend.rund   rB   r   c                 ó   — t        |«      S r   )r   r)   s     r   r+   zSigOptBackend.default_hp_spaceg   rD   r   NrE   r   r   r   rN   rN   ]   rF   r   rN   c                   ó6   — e Zd ZdZed„ «       Zdedefd„Zd„ Z	y)ÚWandbBackendÚwandbc                  ó   — t        «       S r   )r   r   r   r   r   zWandbBackend.is_availablen   s   € ä!Ó#Ð#r   r   r    c                 ó   — t        |||fi |¤ŽS r   )r   r"   s        r   r&   zWandbBackend.runr   s   € Ü" 7¨H°iÑJÀ6ÑJÐJr   c                 ó   — t        |«      S r   )r   r)   s     r   r+   zWandbBackend.default_hp_spaceu   s   € Ü% eÓ,Ð,r   NrE   r   r   r   rT   rT   k   s4   „ Ø€Dàñ$ó ð$ðK Sð K°Só Kó-r   rT   Úreturnc                  ó~  — t         j                  «       D � cg c]  } | j                  «       sŒ| ‘Œ }} t        |«      dkD  rC|d   j                  }t        |«      dkD  r$t
        j                  t        |«      › d|› d�«       |S t        ddj                  d„ t         j                  «       D «       «      z   «      ‚c c} w )Nr   r   z1 hyperparameter search backends available. Using z as the default.z,No hyperparameter search backend available.
ú
c              3   ó\   K  — | ]$  }d |j                   › d|j                  «       › �–— Œ& y­w)z - To install z run N)r   r/   )Ú.0Úbackends     r   ú	<genexpr>z,default_hp_search_backend.<locals>.<genexpr>‰   s3   è ø€ ò 
àð ˜WŸ\™\˜N¨%°×0CÑ0CÓ0EÐ/FÔGñ
ùs   ‚*,)	Ú"ALL_HYPERPARAMETER_SEARCH_BACKENDSÚvaluesr   Úlenr   ÚloggerÚinfor.   Újoin)r^   Úavailable_backendsr   s      r   Údefault_hp_search_backendrg   ~   sÂ   € Ü1S×1ZÑ1ZÓ1\Öw gÐ`g×`tÑ`tÕ`vš'ÐwÐÐwÜ
ÐÓ Ò"Ø! !Ñ$×)Ñ)ˆÜÐ!Ó" QÒ&Ü�K‰KÜÐ)Ó*Ð+Ð+\Ð]aÐ\bÐbrÐsôð ˆÜ
Ø7Ø
�)‰)ñ 
ä=×DÑDÓFô
ó 
ñ	
óð ùò xs
   —B:­B:N)Útypingr   Úintegrationsr   r   r   r   r	   r
   r   r   Útrainer_utilsr   r   r   r   r   Úutilsr   Ú
get_loggerr4   rc   r   r=   rH   rN   rT   r   r`   r7   rg   )r^   s   0r   ú<module>rm      s½   ðõ ÷	÷ 	ó 	÷õ õ ð 
ˆ×	Ñ	˜HÓ	%€÷>ñ >ô2.Ð/ô .ô+Ð0ô +ô.Ð/ô .ô-Ð.ô -ð <IÈ.ÐZgÐiuÐ:vö&Ø/6�O�G—L‘LÓ! 7Ñ*ò&Ð "ð
 3ô ùò&s   Á?B#