jCddlZddlmZmZmZmZmZddlmZddl m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZddlmZmZmZe e gdfZedZedeed eefd Zed d d ed e eegeeffdZ d-d d deeed ed eeee eegeefffdZeedddddfdeeeefdee efdee ge fefdededeededee de d e fdZ defdZ!de"de de d dfdZ#de"de d ee d!fd dfd"Z$de"de d ee d!fd dfd#Z%de"de d ee d!fd dfd$Z&de"de d ee d!fd dfd%Z'de"de d ee d!fd dfd&Z(de"de d ee d!fd dfd'Z)de"de de*d dfd(Z+d)ed efd*Z,de d efd+Z-ee$ee%e.e&e/e'e0e(e1e)iZ2gd,Z3dS).N) _MISSING_TYPEMISSINGFieldfieldfields)wraps)AnyCallableDict ForwardRefListLiteralOptionalTupleTypeTypeVarUnionget_args get_originoverload)#StrictDataclassClassValidationErrorStrictDataclassDefinitionError#StrictDataclassFieldValidationErrorTclsreturncdSN)rs ~/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib/python3.11/site-packages/huggingface_hub/dataclasses.pystrictr"!s%(SF accept_kwargsr%cdSrr r$s r!r"r"%sLOCr#cvdttdttffd }| ||n|S)a Decorator to add strict validation to a dataclass. This decorator must be used on top of `@dataclass` to ensure IDEs and static typing tools recognize the class as a dataclass. Can be used with or without arguments: - `@strict` - `@strict(accept_kwargs=True)` Args: cls: The class to convert to a strict dataclass. accept_kwargs (`bool`, *optional*): If True, allows arbitrary keyword arguments in `__init__`. Defaults to False. Returns: The enhanced dataclass with strict validation on field assignment. Example: ```py >>> from dataclasses import dataclass >>> from huggingface_hub.dataclasses import as_validated_field, strict, validated_field >>> @as_validated_field >>> def positive_int(value: int): ... if not value >= 0: ... raise ValueError(f"Value must be positive, got {value}") >>> @strict(accept_kwargs=True) ... @dataclass ... class User: ... name: str ... age: int = positive_int(default=10) # Initialize >>> User(name="John") User(name='John', age=10) # Extra kwargs are accepted >>> User(name="John", age=30, lastname="Doe") User(name='John', age=30, *lastname='Doe') # Invalid type => raises >>> User(name="John", age="30") huggingface_hub.errors.StrictDataclassFieldValidationError: Validation error for field 'age': TypeError: Field 'age' expected int, got str (value: '30') # Invalid value => raises >>> User(name="John", age=-1) huggingface_hub.errors.StrictDataclassFieldValidationError: Validation error for field 'age': ValueError: Value must be positive, got -1 ``` rrc ztdstdjdi}tD]}g}|t ||jd}|\t|ts|g}|D],}t|std|j d|d-| ||||j <|_ jdtd t d td dffd }|_rdjt%d td dffd }|_jt%d t ffd }|_g} t)D]} | dst-| } t/| s8t1t3j| jdkrtdjd| d| | | _dt:d dffd } d| _tdr.t-jddstdjd| _jt%dfd } tAd| S)N__dataclass_fields__zClass 'z.' must be a dataclass before applying @strict. validatorzInvalid validator for field '': z.. Must be a callable taking a single argument.selfnamevaluerc|j|gD]9} ||#ttf$r}t |||d}~wwxYw|||dS)z1Custom __setattr__ method for strict dataclasses.)rcauseN)__validators__get ValueError TypeErrorr)r,r-r.r*eoriginal_setattrs r!__strict_setattr__z0strict..wrap..__strict_setattr__~s"044T2>> Z Z ZIe$$$$"I.ZZZ=DPQRRRXYYZ  T4 / / / / /s ,AAAkwargscdtDfd|D}|fi||D]\}}|vr||| dS)Nch|] }|j Sr )r-).0fs r! z9strict..wrap..__init__..s#@#@#@qAF#@#@#@r#c$i|] \}}|v || Sr r )r;kvdataclass_fieldss r! z:strict..wrap..__init__..s*"\"\"\DAqaK[F[F[1aF[F[F[r#)ritems __setattr__)r,r8standard_kwargsr-r.rAr original_inits @r!__init__z&strict..wrap..__init__s$A#@F3KK#@#@#@ "\"\"\"\FLLNN"\"\"\ d66o666$*<<>>66KD%#333((u55566r#c|}fd|jD}d|}|r|ddd|dn|S)Nc8g|]\}}|jvd|d|S)*=)r))r;r?r@rs r! z:strict..wrap..__repr__..sC%%%1 888"MMAMM888r#z, ))__dict__rCjoin)r, standard_repradditional_kwargsadditional_reprr original_reprs r!__repr__z&strict..wrap..__repr__s!. d 3 3 %%%%!% 3 3 5 5%%%! #')),=">">GXj-,BBBBBB]jjr# validate_rz' has a class validator 'z' that takes more than one argument. Class validators must take only 'self' as an argument. Methods starting with 'validate_' are considered to be class validators.cjD]>} ||#ttf$r}t|j||d}~wwxYwdS)z%Run class validators on the instance.)r*r0N)__class_validators__r3r4r__name__)r,r*r5rs r!validatez&strict..wrap..validatesz 5 l l lIdOOOO"I.lll= HZbcdddjkkl l ls AAATrZ"__is_defined_by_strict_decorator__Fz' already implements a method called 'validate'. This method name is reserved when using the @strict decorator on a dataclass. If you want to keep your own method, please rename it.cJ|g|Ri||dS)z*Run class validators after initialization.N)rZ)r,argsr8r initial_inits r!init_with_validatez0strict..wrap..init_with_validates= L / / / / / / / LL     r#rG)rN)!hasattrrrYrappend_create_type_validatormetadatar2 isinstancelist _is_validatorr-extendr1rDr strrGrrUdir startswithgetattrcallableleninspect signature parametersrXrr[rZsetattr)rfield_validatorsr< validatorscustom_validatorr*r7rGrUclass_validatorsr-methodrZr_r^rFrTr6r%s` @@@@r!wrapzstrict..wrapcss233 0V#,VVV  :< 2 2AJ   4Q77 8 8 8 z~~k:: +!"2D99:(8'9$!1I(33<AAFAAyAAA!!"2333'1 QV $ $-? 0S 0 0C 0D 0 0 0 0 0 0- ( $LM = ! ! 6 6 6 6 6 6 6 6" ! 6$CL LM = ! ! k# k k k k k k" ! k $CLHH , ,D??;// S$''FF## 7$V,,788A==4>cl>>T>>>  # #F + + + +#3  l1 l l l l l l l7;3 3 # # 3<)MuUU 4NclNNN   | |            Z!3444 r#)rr)rr%rws ` r!r"r")sYtH$q'Hd1gHHHHHHV44999T1r#Tr*defaultdefault_factoryinitreprhashcomparercr8c lt|ts|g}|i}||d<td|||||||d|S)a< Create a dataclass field with a custom validator. Useful to apply several checks to a field. If only applying one rule, check out the [`as_validated_field`] decorator. Args: validator (`Callable` or `List[Callable]`): A method that takes a value as input and raises ValueError/TypeError if the value is invalid. Can be a list of validators to apply multiple checks. **kwargs: Additional arguments to pass to `dataclasses.field()`. Returns: A field with the validator attached in metadata Nr*rxryrzr{r|r}rcr )rdrer) r*rxryrzr{r|r}rcr8s r!validated_fieldrsm4 i & & K %H[  '         r#c6ttdddddfdtttfdttgtftfdt dt dt t dt d t td tffd }|S) a Decorates a validator function as a [`validated_field`] (i.e. a dataclass field with a custom validator). Args: validator (`Callable`): A method that takes a value as input and raises ValueError/TypeError if the value is invalid. TNrxryrzr{r|r}rcr8c .tf|||||||d|S)Nr)r) rxryrzr{r|r}rcr8r*s r!_innerz"as_validated_field.._inner%sC   +      r#)rrr rr boolrr )r*rs` r!as_validated_fieldrs.5CJ##'  sM)* xC0-?@    tn    4.       , Mr#r-r. expected_typec|t|}t|}|turdSt|x}r||||dSt |t rt|||dSt |tst |trdStd|d|)z.Validate that 'value' matches 'expected_type'.NzUnsupported type for field 'r+) rrr _BASIC_TYPE_VALIDATORSr2rdtype_validate_simple_typer rhr4)r-r.roriginr]r*s r!type_validatorr>s  & &F M " "D,0088 8Q $t$$$$$ M4 ( (QdE=99999 M: . .Q*]C2P2PQOtOO OOPPPr#r].c  g}|D]O} t|||dS#t$r,}|t|Yd}~Hd}~wwxYwtd|dt |d|dd|)z8Validate that value matches one of the types in a Union.NField 'z ' with value z doesn't match any type in z . Errors: z; )rr4rarhr{rP)r-r.r]errorstr5s r!_validate_unionrOs F "" " 4 * * * FF " " " MM#a&& ! ! ! ! ! ! ! ! " p$ppT%[[ppTpp]a]f]fgm]n]npp  s A"A  Ac>||vrtd|d|d|dS)zValidate Literal type.rz' expected one of , got N)r4)r-r.r]s r!_validate_literalr^s< DM$MM$MMeMMNNNr#c :t|ts'td|dt|j|d}t |D]F\}} t |d|d||#t$r}td|d|d|d }~wwxYwd S) zValidate List[T] type.rz' expected a list, got r[]Invalid item at index z in list ''N)rdrer4rrY enumeraterr-r.r] item_typeiitemr5s r!_validate_listrds eT " "WU$UUtE{{?SUUVVVQIU##RR4 R d>>Q>>>4 ; ; ; ; R R RIQII$IIIJJPQ Q RRRsA22 B<BBc`t|ts'td|dt|j|\}}|D]W\}} t |d||t |d|d||2#t$r}td|d|d}~wwxYwdS) zValidate Dict[K, V] type.rz' expected a dict, got z.keyrrzInvalid key or value in dict 'rN)rddictr4rrYrCr)r-r.r]key_type value_typer?r@r5s r!_validate_dictrrs eT " "WU$UUtE{{?SUUVVV Hj MM1 M d===!X 6 6 6 d++Q+++Q ; ; ; ; M M MDTDDDEE1 L M MMs+B B+B&&B+c t|ts'td|dt|jt |dkrm|dt ur^t|D]L\}} t|d|d||d$#t$r}td|d |d |d }~wwxYwd St |t |kr2td|d t |d t |tt||D]I\}\}} t|d|d||!#t$r}td|d |d |d }~wwxYwd S)zValidate Tuple type.rz' expected a tuple, got rrrrrz in tuple 'rNz' expected a tuple of length r) rdtupler4rrYrmEllipsisrrzip)r-r.r]rrr5expecteds r!_validate_tuplers eU # #XV$VVU @TVVWWW 4yyA~~$q'X-- '' W WGAt W$~~~~~tT!W==== W W W N N Nt N N NOOUVV W W W Tc%jj b$bbSYYbbVYZ_V`V`bbccc#,S-=-=#>#> W W Ah W$~~~~~tX>>>> W W W N N Nt N N NOOUVV W W Ws04B B8B33B85E  E3E..E3c.t|ts'td|dt|j|d}t |D]@\}} t |d||#t$r}td|d|d}~wwxYwdS)zValidate Set[T] type.rz' expected a set, got rz itemzInvalid item in set 'rN)rdsetr4rrYrrrs r! _validate_setrs eS ! !VT$TTd5kk>RTTUUUQIU##DD4 D d>>>4 ; ; ; ; D D D;D;;;<tj|jdSr)rr-r)r.rs r!r*z)_create_type_validator..validatorsuz5%*55555r#)r )rr*s` r!rbrbs766666666 r#ct|sdStj|}t|j}t |dkrdS|djtjj tjj tjj fvrdS|ddD]}|j tjj krdS dS)zCheck if a function is a validator. A validator is a Callable that can be called with a single positional argument. The validator can have more arguments with default values. Basically, returns True if `validator(value)` is possible. FrrNT)rlrnrorerpvaluesrmkind ParameterPOSITIONAL_OR_KEYWORDPOSITIONAL_ONLYVAR_POSITIONALrxempty)r*rorp parameters r!rfrfs I  u!),,Ii*113344J :!u!}/)(" u^   1 7 7 755 8 4r#)r"r Validator_Trrrr)4rn dataclassesrrrrr functoolsrtypingr r r r r rrrrrrrrrrrrrrrr"rrrrhrrrrrrrrrrbrfrerrrr__all__r r#r!rs`DDDDDDDDDDDDDD"ud{#  GCLL (Q(DG((( ( $)OOOTOhQy$q'7I.JOOO O"E2