o ci @sddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZd d Zd d ZddZddZddZddZddZddZddZeeeeeeeeeed Zedddd Zed!Zed"Zed#Z gd$Z!d%S)&)get_function_argsBasicVariantGenerator)ConcurrencyLimiter)Repeater)SearchAlgorithm)SearchGenerator)Searcher) grid_search) PublicAPIcCstS)Nrr r L/home/ubuntu/.local/lib/python3.10/site-packages/ray/tune/search/__init__.py_import_variant_generator srcCddlm}|S)NrAxSearch)ray.tune.search.ax.ax_searchrrr r r _import_ax_search rcCr)NrHyperOptSearch)(ray.tune.search.hyperopt.hyperopt_searchrrr r r _import_hyperopt_searchrrcCr)NrBayesOptSearch)(ray.tune.search.bayesopt.bayesopt_searchrrr r r _import_bayesopt_searchrrcCr)NrTuneBOHB) ray.tune.search.bohb.bohb_searchrrr r r _import_bohb_search"rr cCr)NrNevergradSearch)*ray.tune.search.nevergrad.nevergrad_searchr"r!r r r _import_nevergrad_search(rr$cCr)Nr OptunaSearch)$ray.tune.search.optuna.optuna_searchr&r%r r r _import_optuna_search.rr(cCr)Nr ZOOptSearch)"ray.tune.search.zoopt.zoopt_searchr*r)r r r _import_zoopt_search4rr,cCr)Nr HEBOSearch) ray.tune.search.hebo.hebo_searchr.r-r r r _import_hebo_search:rr0) variant_generatorrandomaxhyperoptbayesoptbohb nevergradoptunazoopthebobeta) stabilityc s^|}|tvrtdttd|t|}t|fdd|D}|di|S)aInstantiate a search algorithm based on the given string. This is useful for swapping between different search algorithms. Args: search_alg: The search algorithm to use. metric: The training result objective value attribute. Stopping procedures will use this attribute. mode: One of {min, max}. Determines whether objective is minimizing or maximizing the metric attribute. **kwargs: Additional parameters. These keyword arguments will be passed to the initialization function of the chosen class. Returns: ray.tune.search.Searcher: The search algorithm. Example: >>> from ray import tune # doctest: +SKIP >>> search_alg = tune.create_searcher('ax') # doctest: +SKIP z)The `search_alg` argument must be one of z. Got: csi|] \}}|vr||qSr r ).0kvsearch_alg_argsr r rsz#create_searcher..Nr )lowerSEARCH_ALG_IMPORT ValueErrorlistritems) search_algkwargs SearcherClasstrimmed_kwargsr r@r create_searcherNs rLa:You passed a `{par}` parameter to {cls} that contained unresolved search space definitions. {cls} should however be instantiated with fully configured search spaces only. To use Ray Tune's automatic search space conversion, pass the space definition as part of the `param_space` argument to `tune.Tuner()` instead.zTrying to sample a configuration from {cls}, but no search space has been defined. Either pass the `{space}` argument when instantiating the search algorithm, or pass a `param_space` to `tune.Tuner()`.zTrying to sample a configuration from {cls}, but the `metric` ({metric}) or `mode` ({mode}) parameters have not been set. Either pass these arguments when instantiating the search algorithm, or pass them to `tune.TuneConfig()`.) rr rrrr rUNRESOLVED_SEARCH_SPACEUNDEFINED_SEARCH_SPACEUNDEFINED_METRIC_MODEN)"ray._common.utilsrray.tune.search.basic_variantr#ray.tune.search.concurrency_limiterrray.tune.search.repeaterr ray.tune.search.search_algorithmr ray.tune.search.search_generatorrray.tune.search.searcherr !ray.tune.search.variant_generatorr ray.utilr rrrrr r$r(r,r0rDrLstrrMrNrO__all__r r r r sR         (