o /wi@sddlmZddlmZddlZddlZddlmZddl m Z ddl m Z ddl mZddl mZdddZdddZddd ddZdS)!) annotations)SequenceN)_CONSTRAINTS_KEY)StudyDirection) _dominates) FrozenTrial) TrialStatetrial0rtrial1 directionsSequence[StudyDirection]returnboolc Csd|jt}|jt}|durtd|jd|dur(td|jd|dur6|dur6t|||S|dur@|dur@dS|durJ|durJdSt|tt fsSJt|tt fs\Jt |t |krht d|j t jkrpdS|j t jkrxdStdd|D}td d|D}|r|rt|||S|rdS|rdStd d|D}td d|D}||kS) afChecks constrained-domination. A trial x is said to constrained-dominate a trial y, if any of the following conditions is true: 1) Trial x is feasible and trial y is not. 2) Trial x and y are both infeasible, but solution x has a smaller overall constraint violation. 3) Trial x and y are feasible and trial x dominates trial y. NTrial K does not have constraint values. It will be dominated by the other trials.TF@Trials with different numbers of constraints cannot be compared.cs|]}|dkVqdSrN.0vrrk/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/optuna/samplers/nsgaii/_constraints_evaluation.py Cz)_constrained_dominates..csrrrrrrrrDrcs|] }|dkr|VqdSrrrrrrrScsrrrrrrrrTr) system_attrsgetrwarningswarnnumberr isinstancelisttuplelen ValueErrorstaterCOMPLETEallsum) r r r constraints0 constraints1satisfy_constraints0satisfy_constraints1 violation0 violation1rrr_constrained_dominatessF       r1 populationSequence[FrozenTrial] np.ndarraycCsRg}|D]}|jt}|dur|tjq|tdd|Dqt|S)zEvaluate feasibility of trials in population. Returns: A list of feasibility status T/F/None of trials in population, where T/F means feasible/infeasible and None means that the trial does not have constraint values. Ncsrrrrrrrrerz$_evaluate_penalty..)rrrappendnpnanr*array)r2penaltytrial constraintsrrr_evaluate_penaltyXs  r<F)is_constrainedlist[FrozenTrial]r=NonecCs|sdStdd|Ddd}|D]0}|jt}|dur(td|jdqtt t |r7t dt ||krAt dqdS) NcSsg|] }t|jtgqSr)r%rrr)rtrrr rsz)_validate_constraints..r)defaultrrz*NaN is not acceptable as constraint value.r) maxrrrrr r!r6anyisnanr8r&r%)r2r=num_constraints_trial _constraintsrrr_validate_constraintsis$   rI)r rr rr r r r)r2r3r r4)r2r>r=rr r?) __future__rcollections.abcrrnumpyr6optuna.samplers._baser optuna.studyroptuna.study._multi_objectiver optuna.trialrrr1r<rIrrrrs        I