o ,wi. @srddlZddlZddlmZddlmZmZmZmZmZm Z ddl m Z ddl m Z mZmZmZmZmZmZe eeeeeefZdeeefdedefd eeeffd d Zde dedefd e fd d Zdededefd efddZdededefd efddZe eefZded e fddZ dedeed efddZ!dddedeed efddZ"dddedeed efddZ#dddedeed efddZ$dddedeed efddZ%de eeeeeeefeee fd e fd d!Z&  "d>d#e eefd$ee eefd%e eefd efd&d'Z'd#e eefd(e eefd efd)d*Z(dd+de ee efd,eed efd-d.Z)ddd/de ee efd,ee e efd0eeed e eee effd1d2Z*ddd3d4de ee efd,ee e efd0eeed5ed ef d6d7Z+d,e e efd5ed e e effd8d9Z, d?d:e eeefd;ee eeefd efdg}|jD]}||}t|tsJ||qt|j|dS)N) simple_formlist)rr ElementTypeappendr r)rrchoicesitemchoicerrr cast_choice!s   r"cCs t||j||jt|jdS)N)startendtags)r r#r$rr%rrrrr cast_interval*sr'cCs4|ttfvr tdt||j||j||jdS)Nz&Range can only be cast to int or floatr#stopstep) cast_floatcast_int ValueErrorrr#r)r*r&rrr cast_range0s r.argscGs8g}|D]}t|ts J||qtt|ddS)NT)rr)r r rr builtinsr)r/rargrrr_list_to_simple_choice=s  r2cGsht|dkr|durtd|dur|St|dkrtdt|dkr(|dSt|dkr2t|SJ)Nrz,cannot use both position and named argumentsz%No positional args or value specified)len TypeErrorr2rr/rrr_normalize_cast_valueEs   r7)rcGst|d|i}t|trt|jSt|trt|tSt|tr'ttt|St|t r1t |tSt|t r;t |tSt|t rEt|tSt|ttttfsPJt|SNr)r7 isinstancerr,textdictrrmapr r"rr.r r'intfloatboolstrr6rrrr,S           r,cGst|d|i}t|trt|jSt|trt|tSt|tr'ttt|St|t r1t |tSt|t r;t |tSt|t rEt|tSt|ttttfsPJt|Sr8)r7r9rr+r:r;rrr<r r"rr.r r'r=r>r?r@r6rrrr+erAr+cGst|d|i}t|trt|jSt|trt|tSt|tr'ttt|St|t r1t |tSt|t r;t |tSt|t rDtdt|ttttfsOJt|trZt|St|S)NrzIntervals cannot be cast to str)r7r9rcast_strr:r;rrr<r r"rr.r r-r=r>r?r@lowerr6rrrrBws"            rBcGst|d|i}t|trt|jSt|trt|tSt|tr'ttt|St|t r1t |tSt|t r;t |tSt|t rDtdt|tra|dkrQdS|dkrYdStd|dt|S) Nrz Intervals cannot be cast to boolfalseFtrueTz Cannot cast 'z ' to bool)r7r9r cast_boolr:r;rrr<r r"rr.r r-r@rCr?r6rrrrFs(             rFcGsXt|dkr tdt|dkr%|d}t|tr%|jr!d|_|Stdtt|dS)z2 A choice sweep over the specified values rzempty choice is not legalr3Fz nesting choices is not supported)r)r4r-r9r rr)r/firstrrrr!s   r!r3r#r)r*cCs|dur|}d}t|||dS)a Range defines a sweep over a range of integer or floating-point values. When only start is defined, it is set as the stop value, and start is set at zero. For a positive step, the contents of a range r are determined by the formula r[i] = start + step*i where i >= 0 and r[i] < stop. For a negative step, the contents of the range are still determined by the formula r[i] = start + step*i, but the constraints are i >= 0 and r[i] > stop. Nrr()rr(rrrrangesrHr$cCstt|t|dS)z{ A continuous interval between two floating point values. value=interval(x,y) is interpreted as x <= value < y r#r$)r r>rIrrrintervalsrJ)sweeprKcGst|dkr td|durtt||gS|d}t|trG|}t}|ddD]}t|ts|d}t|ttfr*t|}d|_|St|tjr;t|}t||S|gSt |}d|_|S)zA Shuffle input list or sweep (does not support interval) Nr3rT) shuffler4r9r rrr0rrandomr2)rKrr/r1lst simple_choicerrrrTs$   rTF)rKrreverserXcGs|dur t||dS|durt||St|dkrC|d}t|ttfr)t||St|tjr5t||dSt |t r<|St d|t t ttf}|D]}t||sYt d|qKt|dkrdtdt|dkrst|}t||SdS)zJ Sort an input list or sweep. reverse=True reverses the order NrXr3rzInvalid arguments: zempty sort input)sort _sort_sweepr4r9r rr0rsortedr r r5r=r>r?r@r-r2)rKrrXr/r1 primitivescwrrrrZs0            rZcCst|}t|trt|j|d|_|St|trn|jdusJ|jdus&J|sK|j|jkrI|jt|j }|jt|j }||_||_|j |_ |S|j|jkrl|jt|j }|jt|j }||_||_|j |_ |SJ)NrY) rr9r r\rrr#r)absr*)rKrXr#r)rrrr[9s.      r[includeexcludecCs:t|tr|g}|durg}nt|tr|g}t||dS)a A glob selects from all options in the config group. inputs are in glob format. e.g: *, foo*, *foo. :param include: a string or a list of strings to use as include globs :param exclude: a string or a list of strings to use as exclude globs Nr`ra)r9r@r rbrrrglobZs   rc)Nr3r).r0rUrtypingrrrrrrhydra._internal.grammar.utilsr hydra.core.override_parser.typesr r r r rrrr@r=r?r>rr;rrr"r'r.CastTyper2r7r,r+rBrFr!rHrJrMrTrZr[rcrrrrs   $                 &,   !   (   "