L ibPddlmZddlmZmZmZmZmZmZm Z m Z ddl m Z m Z mZmZmZddlmZej&eZGddZGdd eZGd d eZGd d eZGddeZeeeefDcic]}e |j6|c}ZdefdZycc}w))Optional)is_optuna_availableis_ray_tune_availableis_sigopt_availableis_wandb_availablerun_hp_search_optunarun_hp_search_rayrun_hp_search_sigoptrun_hp_search_wandb)HPSearchBackenddefault_hp_space_optunadefault_hp_space_raydefault_hp_space_sigoptdefault_hp_space_wandb)loggingcheZdZUeed<dZeeed<edZde defdZ dZ d Z e d Zy) HyperParamSearchBackendBasenameN pip_packagectNNotImplementedErrorh/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/transformers/hyperparameter_search.py is_availablez(HyperParamSearchBackendBase.is_available+s!!rn_trials directionc trrselftrainerrr kwargss rrunzHyperParamSearchBackendBase.run/!!rctrrr#trials rdefault_hp_spacez,HyperParamSearchBackendBase.default_hp_space2r'rcx|js*td|jd|jdy)NzYou picked the z' backend, but it is not installed. Run .)r RuntimeErrorr pip_install)r#s rensure_availablez,HyperParamSearchBackendBase.ensure_available5sC  "!$)),STXTdTdTfSgghi #rc>d|jxs |jdS)Nz `pip install `)rr)clss rr/z'HyperParamSearchBackendBase.pip_install;ss:#((;1==r)__name__ __module__ __qualname__str__annotations__rr staticmethodrintr&r+r0 classmethodr/rrrrr'sX I!%K#%"""S"S"" >>rrc6eZdZdZedZdedefdZdZ y) OptunaBackendoptunactSr)rrrrrzOptunaBackend.is_availableC "$$rrr c t|||fi|Sr)r r"s rr&zOptunaBackend.runG#GXyKFKKrct|Sr)rr)s rr+zOptunaBackend.default_hp_spaceJ &u--rN r4r5r6rr9rr:r7r&r+rrrr=r=@4 D%%LSLSL.rr=c:eZdZdZdZedZdedefdZ dZ y) RayTuneBackendrayz 'ray[tune]'ctSr)rrrrrzRayTuneBackend.is_availableRs $&&rrr c t|||fi|Sr)r r"s rr&zRayTuneBackend.runVs (IHHHrct|Sr)rr)s rr+zRayTuneBackend.default_hp_spaceYs #E**rN) r4r5r6rrr9rr:r7r&r+rrrrHrHNs9 DK''ISISI+rrHc6eZdZdZedZdedefdZdZ y) SigOptBackendsigoptctSr)rrrrrzSigOptBackend.is_available`r@rrr c t|||fi|Sr)r r"s rr&zSigOptBackend.rundrBrct|Sr)rr)s rr+zSigOptBackend.default_hp_spacegrDrNrErrrrNrN]rFrrNc6eZdZdZedZdedefdZdZ y) WandbBackendwandbctSr)rrrrrzWandbBackend.is_availablens !##rrr c t|||fi|Sr)r r"s rr&zWandbBackend.runrs"7HiJ6JJrct|Sr)rr)s rr+zWandbBackend.default_hp_spaceus %e,,rNrErrrrTrTks4 D$$KSKSK-rrTreturnc~tjDcgc]}|js|}}t|dkDrC|dj}t|dkDr$t j t|d|d|StddjdtjDzcc}w)Nrrz1 hyperparameter search backends available. Using z as the default.z,No hyperparameter search backend available.  c3\K|]$}d|jd|j&yw)z - To install z run N)rr/).0backends r z,default_hp_search_backend..s3 W\\N%0C0C0E/F G s*,) "ALL_HYPERPARAMETER_SEARCH_BACKENDSvaluesrlenrloggerinfor.join)r^available_backendsrs rdefault_hp_search_backendrg~s1S1Z1Z1\wg`g`t`t`v'ww "!!$)) ! "Q & KK)*++\]a\bbrs  7 )) =DDF    xs B:B:N)typingr integrationsrrrrr r r r trainer_utilsr rrrrutilsr get_loggerr4rcrr=rHrNrTrr`r7rg)r^s0rrms      H %>>2 ./ . +0 + ./ . -. -