o }oi@sddlmZddlmZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdddZd ddZddd!ddZdS)") annotations)SequenceN) optuna_warn)_CONSTRAINTS_KEY)StudyDirection) _dominates) FrozenTrial) TrialStatetrial0rtrial1 directionsSequence[StudyDirection]returnboolc Cs`|jt}|jt}|durtd|jd|dur&td|jd|dur4|dur4t|||S|dur>|dur>dS|durH|durHdSt|ttfsQJt|ttfsZJt |t |krft d|j t j krndS|j t j krvdStdd|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.0vrrb/home/ubuntu/.local/lib/python3.10/site-packages/optuna/samplers/nsgaii/_constraints_evaluation.py Cz)_constrained_dominates..csrrrrrrrrDrcs|] }|dkr|VqdSrrrrrrrScsrrrrrrrrTr) system_attrsgetrrnumberr isinstancelisttuplelen ValueErrorstater COMPLETEallsum) r r r constraints0 constraints1satisfy_constraints0satisfy_constraints1 violation0 violation1rrr_constrained_dominatessF       r0 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)r1penaltytrial constraintsrrr_evaluate_penaltyXs  r;F)is_constrainedlist[FrozenTrial]r<NonecCs|sdStdd|Ddd}|D]/}|jt}|dur'td|jdqttt |r6t dt ||kr@t dqdS) NcSsg|] }t|jtgqSr)r$rrr)rtrrr rsz)_validate_constraints..r)defaultrrz*NaN is not acceptable as constraint value.r) maxrrrrr r5anyisnanr7r%r$)r1r<num_constraints_trial _constraintsrrr_validate_constraintsis$   rH)r rr rr r rr)r1r2rr3)r1r=r<rrr>) __future__rcollections.abcrnumpyr5optuna._warningsroptuna.samplers._baser optuna.studyroptuna.study._multi_objectiver optuna.trialrr r0r;rHrrrrs         I