-jidSSKJr SSKrSSKJr SSKrSSKJr SSKJ r S Sjr "SS\5r g) ) annotationsN)Any) BasePruner)_is_first_in_interval_stepc[U5nU$![[4a& S[U5RS3n[U5Sef=f)Nz!The `value` argument is of type 'z' but supposed to be a float.)float TypeError ValueErrortype__name__)valuemessages S/home/james-whalen/.local/lib/python3.13/site-packages/optuna/pruners/_threshold.py _check_valuer sY+e  L z "+/U 0D0D/EEb c  d* +s 6AcJ\rSrSrSrSSSjjrS SjrSrg) ThresholdPrunera2Pruner to detect outlying metrics of the trials. Prune if a metric exceeds upper threshold, falls behind lower threshold or reaches ``nan``. Example: .. testcode:: from optuna import create_study from optuna.pruners import ThresholdPruner from optuna import TrialPruned def objective_for_upper(trial): for step, y in enumerate(ys_for_upper): trial.report(y, step) if trial.should_prune(): raise TrialPruned() return ys_for_upper[-1] def objective_for_lower(trial): for step, y in enumerate(ys_for_lower): trial.report(y, step) if trial.should_prune(): raise TrialPruned() return ys_for_lower[-1] ys_for_upper = [0.0, 0.1, 0.2, 0.5, 1.2] ys_for_lower = [100.0, 90.0, 0.1, 0.0, -1] study = create_study(pruner=ThresholdPruner(upper=1.0)) study.optimize(objective_for_upper, n_trials=10) study = create_study(pruner=ThresholdPruner(lower=0.0)) study.optimize(objective_for_lower, n_trials=10) Args: lower: A minimum value which determines whether pruner prunes or not. If an intermediate value is smaller than lower, it prunes. upper: A maximum value which determines whether pruner prunes or not. If an intermediate value is larger than upper, it prunes. n_warmup_steps: Pruning is disabled if the step is less than the given number of warmup steps. interval_steps: Interval in number of steps between the pruning checks, offset by the warmup steps. If no value has been reported at the time of a pruning check, that particular check will be postponed until a value is reported. Value must be at least 1. NcDUcUc [S5eUb [U5nUb [U5nUbUO [S5*nUbUO [S5nX:a [S5eUS:a[SUS35eUS:a[SUS35eXlX lX0lX@lg) Nz(Either lower or upper must be specified.infz#lower should be smaller than upper.rz2Number of warmup steps cannot be negative but got .z2Pruning interval steps must be at least 1 but got )r rrr _lower_upper_n_warmup_steps_interval_steps)selfloweruppern_warmup_stepsinterval_stepss r__init__ThresholdPruner.__init__Qs =U]FG G   'E   'E*u *e  =BC C A D^DTTUV  A D^DTTUV   --cFURnUcgURnX4:ag[X2RR 5X@R 5(dgURUn[ R"U5(agXPR:agXPR:agg)NFT) last_steprrintermediate_valueskeysrmathisnanrr)rstudytrialstepr latest_values rpruneThresholdPruner.pruners <--  ) ++002NDXDX  006 ::l # # ++ % ++ %r#)rrrr)NNrr) r float | Nonerr0rintr r1returnNone)r*z'optuna.study.Study'r+z'optuna.trial.FrozenTrial'r2bool)r __module__ __qualname____firstlineno____doc__r!r.__static_attributes__r#rrrsQ6t#" ... .  .  .Br#r)r rr2r) __future__rr(typingroptunaoptuna.prunersroptuna.pruners._percentilerrrr:r#rr@s*"  %A rjrr#