-ji-BSSKJr SSKJr SSKrSSKJr SSKJr SSKr SSK J r SSK J r SSK Jr SS K Jr SS K Jr SS KJr SS KJr SS KJr SSKJr SSKJr \(aSSKJr SSKJr SSKJr \"SS55r\ "S5"SS\55rSSjr g)) annotations) dataclassN)Any) TYPE_CHECKING)experimental_class)BaseDistribution)CategoricalDistribution)FloatDistribution)IntDistribution) BaseSampler)LazyRandomState) create_trial) FrozenTrial) TrialState)Iterable)Sequence)Studyc\rSrSr%SrS\S'SrS\S'SrS\S 'SS jrSS jr SS jr SS jr SSjr SSjr Srg) _TreeNodeN str | None param_namezdict[float, '_TreeNode'] | NonechildrenFbool is_runningcURc'XlUVs0sH o3[5_M snUlgURU:wa[SURSU35eURR 5[ U5:wa;[S[ URR 55S[ U535egs snf)Nzparam_name mismatch: z != zsearch_space mismatch: )rrr ValueErrorkeysset)selfr search_spacevalues V/home/james-whalen/.local/lib/python3.13/site-packages/optuna/samplers/_brute_force.pyexpand_TreeNode.expand*s == (O=IJ\EIK/\JDM*, #88Ij\!Z[[}}!!#s<'88 -c$--2D2D2F.G-HSQ]M^L_`9 KsCcSUlg)NTrr s r# set_running_TreeNode.set_running9s c(URS/5 gN)r$r(s r#set_leaf_TreeNode.set_leaf<s D"r+cUnUHFup4nURX45 URceXRR;a gURUnMH U$r-)r$r)r params_and_search_spaces current_noderr!r"s r#add_path_TreeNode.add_path?s` /G +Je    9((4 44111'007L 0H r+c^URcT(aUR(aS$S$[U4SjURR555$)Nrc3D># UHoRT5v M g7fr-)count_unexpanded).0childexclude_runnings r# -_TreeNode.count_unexpanded..QscLb5--o>>Lbs )rrsumvalues)r r;s `r#r8_TreeNode.count_unexpandedLs> == 'DOO1 B BcDMML`L`Lbcc cr+cf^URce[R"URR5Vs/sHo3R U5PM sn[R S9m[ U4Sj[URR5555(aD[URR55HupCUR(dMSTU'M TTR5-mUR[URR55TS9$s snf)N)dtypec3h># UH'upUR(+=(a TUS:v M) g7f)rNr')r9ir"weightss r#r<)_TreeNode.sample_child..]s3 =  3WQZ!^ 3=s/2g)p) rnparrayr?r8float64any enumeraterr>choicelistr)r rngr;r:rDrEs @r# sample_child_TreeNode.sample_childSs}}((( ((BF--BVBVBX YBX # #O 4BX Y**   %dmm&:&:&<=   &dmm&:&:&<=###!$GAJ> 7;;= zz$t}}1134z@@ ZsD.)rrr)rrr!zIterable[float]returnNone)rRrS)r1z,Iterable[tuple[str, Iterable[float], float]]rRz_TreeNode | None)r;rrRint)rOznp.random.RandomStater;rrRfloat)__name__ __module__ __qualname____firstlineno__r__annotations__rrr$r)r.r3r8rP__static_attributes__r+r#rrsS"J !04H-4J  (T   dAr+rz3.1.0c\rSrSrSrS S SjjrS SjrSSjr\SSj5r SSjr SS jr S r g)BruteForceSampleriaSampler using brute force. This sampler performs exhaustive search on the defined search space. Example: .. testcode:: import optuna def objective(trial): c = trial.suggest_categorical("c", ["float", "int"]) if c == "float": return trial.suggest_float("x", 1, 3, step=0.5) elif c == "int": a = trial.suggest_int("a", 1, 3) b = trial.suggest_int("b", a, 3) return a + b study = optuna.create_study(sampler=optuna.samplers.BruteForceSampler()) study.optimize(objective) Note: The defined search space must be finite. Therefore, when using :class:`~optuna.distributions.FloatDistribution` or :func:`~optuna.trial.Trial.suggest_float`, ``step=None`` is not allowed. Note: The sampler may fail to try the entire search space in when the suggestion ranges or parameters are changed in the same :class:`~optuna.study.Study`. Args: seed: A seed to fix the order of trials as the search order randomly shuffled. Please note that it is not recommended using this option in distributed optimization settings since this option cannot ensure the order of trials and may increase the number of duplicate suggestions during distributed optimization. avoid_premature_stop: If :obj:`True`, the sampler performs a strict exhaustive search. Please note that enabling this option may increase the likelihood of duplicate sampling. When this option is not enabled (default), the sampler applies a looser criterion for determining when to stop the search, which may result in incomplete coverage of the search space. For more information, see https://github.com/optuna/optuna/issues/5780. Nc0[U5UlX lgr-)r _rng_avoid_premature_stop)r seedavoid_premature_stops r#__init__BruteForceSampler.__init__s#D) %9"r+c0$r-r\)r studytrials r#infer_relative_search_space-BruteForceSampler.infer_relative_search_space  r+c0$r-r\)r rhrir!s r#sample_relative!BruteForceSampler.sample_relativerlr+cb^^UHm[U4SjTR555(dM-URUU4SjTRR555nUcMfTRR 5(aUR 5 MUR5 M g)Nc3x># UH/upUTR;=(a TRUU:Hv M1 g7fr-)params)r9rGvris r#r<3BruteForceSampler._populate_tree..s1]ndaqELL(AU\\!_-AAns7:c3># UH9upUT;dM U[U5URTRU54v M; g7fr-)_enumerate_candidatesto_internal_reprrr)r9rparam_distributionrrris r#r<rtsQ ;V6 !/ "-.@A*;;ELL# UH"oRTR:wdMUv M$ g7fr-)number)r9tris r#r<7BruteForceSampler.sample_independent..s "QfELL8P11fs- -r)rb_storageget_all_trials _study_idrCOMPLETEPRUNEDRUNNINGFAILrrvr$rrrr8to_external_reprrarOrMrP) r rhrirrxr;rr~ candidatess ` r#sample_independent$BruteForceSampler.sample_independents#888 .. OO##!!"" /  {*+=>  J+ D"Qf"QSXS_S_`   1Q 6%66tyy}}7K7KJ7WX X%66!!$))--A r+c^^^UR(+nURRURS[R [R [R[R4S9n[5nURUUUU4SjU505 URU5S:XaUR5 gg)NFrc3># UHAnURTR:waUO[TTTRTRS9v MC g7f))r|r?rrr{N)rrrrr{)r9rr|rir?s r#r<0BruteForceSampler.after_trial..sQ  Axx5<</%#%$||&+&9&9  sA A r) rbrrrrrrrrrrr8stop)r rhrir|r?r;rr~s ``` r# after_trialBruteForceSampler.after_trials#888 .. OO##!!"" /  {       $   1Q 6 JJL 7r+)rbra)NF)rcz int | NonerdrrRrS)rhrrirrRdict[str, BaseDistribution])rhrrirr!rrRdict[str, Any])r~rrzIterable[FrozenTrial]rrrrRrS) rhrrirrstrrxrrRr) rhrrirr|rr?zSequence[float] | NonerRrS) rVrWrXrY__doc__rerjrn staticmethodrrrr[r\r+r#r^r^is-^:#. $ #.>Y  ''!6'@N' ''2""" " - " "H*** * ' *  *r+r^c[U[5(aURc [S5e[R "[ UR55n[R "[ UR55n[R "[ UR55n/nUnXR::a%UR[U55 XS- nXR::aM%U$[U[5(a7[[URURS-UR55$[U[5(a'[[[UR 555$[SUS35e)NzjFloatDistribution.step must be given for BruteForceSampler (otherwise, the search space will be infinite).r6zUnknown distribution .) isinstancer steprdecimalDecimalrlowhighappendrUr rNranger lenchoices)rxrrrretr"s r#rvrvs9$&788  " " *C ooc"4"8"89:s#5#:#:;<s#5#:#:;<m JJuU| $ MEm & 8 8 $((*<*A*AA*EGYG^G^ _   &(? @ @E#0889:;;01C0DAFGGr+)rxrrRzSequence[float])! __future__r dataclassesrrtypingrrnumpyrHoptuna._experimentalroptuna.distributionsrr r r optuna.samplersr "optuna.samplers._lazy_random_stater optuna.trialrrrcollections.abcrr optuna.studyrrr^rvr\r+r#rs"! 31820'>%$#((" HAHA HAVGf ffRHr+