3jSSKJr SSKJrJr SSKJrJr SSKJ r SSK r \ (aSSK J r /SQr "SS \5r\"S S 9"S S \55r\"S S 9"SS\55r\"S S 9"SS\55r\"S S 9"SS55r\"S S 9"SS55r\"SS55rSSjrSSSjjrg)) annotations)ABCabstractmethod) dataclassfield) TYPE_CHECKINGN)Mapping) DivisibleBy ExactDimsFunctionConstraintsMinDimsParamConstraint ShapeRuleValidationResultvalidate_function_callvalidate_paramc@\rSrSrSr\SSj5r\SSj5rSrg) rz&Base class for shape validation rules.cg)z*Check if tensor satisfies this shape rule.Nselftensors S/home/wildlama/miniconda3/lib/python3.13/site-packages/comfy_kitchen/constraints.pycheckShapeRule.check! cg)z'Human-readable description of the rule.Nrrs rdescribeShapeRule.describe&rrrNr torch.Tensorreturnboolr%str) __name__ __module__ __qualname____firstlineno____doc__rrr!__static_attributes__rrrrrs+0    rrT)frozencB\rSrSr%SrS\S'S\S'S SjrS SjrSrg ) r ,z7Check that a tensor dimension is divisible by a factor.intdimfactorcUR5UR::agURURUR-S:H$)NFr)r3shaper4rs rrDivisibleBy.check3s8 ::<488 #||DHH% 3q88rc:SURSUR3$)Nzdim[z] divisible by )r3r4r s rr!DivisibleBy.describe8sdhhZt{{m<rs rr MinDims.checkBzz|tyy((rc"SURS3$)Nz at least Dr>r s rr!MinDims.describeEs499+Q''rrNr#r'r:rrrr r <s6 I)(rr c8\rSrSr%SrS\S'S SjrS SjrSrg) r Iz+Check that tensor has exactly N dimensions.r2r>c<UR5UR:H$r@rArs rrExactDims.checkOrCrc"SURS3$)Nzexactly rErFr s rr!ExactDims.describeRs$))A&&rrNr#r'r:rrrr r Is5 I)'rr cb\rSrSr%SrSrS\S'SrS\S'SrS \S 'SS jr SS jr SS jr Sr g)r[zConstraints for a single parameter. Attributes: dtypes: Allowed dtypes for this parameter. None means any dtype. devices: Allowed device types. None means inherit from function default. shape_rules: Tuple of shape rules that must all pass. Nzfrozenset[torch.dtype] | Nonedtypeszfrozenset[str] | Nonedevicesrztuple[ShapeRule, ...] shape_rulescURcg[U[R5(aURUR;$XR;$)z"Check if value's dtype is allowed.T)rP isinstancetorchTensordtype)rvalues r check_dtypeParamConstraint.check_dtypeis@ ;;  eU\\ * *;;$++- - ##rcxURb UROUnSU;agURRU;$)z$Check if tensor's device is allowed.*T)rQdevicetype)rrdefault_devicesrQs r check_deviceParamConstraint.check_devicers5"&,,":$,, '>}}!!W,,rcB^[U4SjUR55$)z*Check if tensor satisfies all shape rules.c3D># UHoRT5v M g7fr@)r).0rulers r .ParamConstraint.check_shape..{sC2B$::f%%2Bs )allrRrs `r check_shapeParamConstraint.check_shapeysC$2B2BCCCr)rXztorch.Tensor | torch.dtyper%r&)rr$r_frozenset[str]r%r&r#) r)r*r+r,r-rPr;rQrRrYr`rir.rrrrr[s:-1F )0%)G "))+K&+$-Drrc`\rSrSr%Sr\"\S9rS\S'\"SS9r S\S'S r S \S 'SS jr S r g )r ~aConstraints for an entire function. Attributes: params: Per-parameter constraints keyed by parameter name. default_devices: Default allowed devices when param doesn't specify. min_compute_capability: Minimum CUDA compute capability (major, minor). )default_factoryzdict[str, ParamConstraint]paramsc[SS15$)Ncudacpu) frozensetrrrFunctionConstraints.sIvW\oD^rrkr_Ntuple[int, int] | Nonemin_compute_capabilityc [[[URR 5SS95UR UR 45$)Nc US$)Nrr)xs rrt.FunctionConstraints.__hash__..sAaDr)key)hashtuplesortedroitemsr_rwr s r__hash__FunctionConstraints.__hash__sB &**,.A B   ' '   rr)r%r2) r)r*r+r,r-rdictror;r_rwrr.rrrr r ~s:*/t)DF &D&+<^&_O^_5929 rr ch\rSrSr%SrS\S'SrS\S'SrS\S'\S S j5r \S S j5r S r g)rz Result of constraint validation.r&successNz str | None failed_paramfailure_reasonc[SS9$)NT)rrrrrokValidationResult.oks --rc[SXS9$)NF)rrrr)paramreasons rfailValidationResult.failsEYYrr)r%r)rr(rr(r%r) r)r*r+r,r-r;rr staticmethodrrr.rrrrrsF* M#L*#!%NJ%..ZZrrc Uc[R5$URU5(d[U[R 5(a9[R USURS[UR535$[R USUS[UR535$[U[R 5(d[R5$URX5(dTURb UROUn[R USURRS[U535$URHXnURU5(aM US[!UR"5SUR%535s $ [R5$)a?Validate a single parameter against its constraint. Args: name: Parameter name (for error messages) value: The parameter value constraint: Constraint to validate against default_devices: Default devices from FunctionConstraints Returns: ValidationResult indicating success or failure with details zdtype z not in zdevice zshape z fails: )rrrYrTrUrVrrWsetrPr`rQr]r^rRrlistr6r!)namerX constraintr_allowedres rrrs" }""$$  ! !% ( ( eU\\ * *#((u{{m8C 8I8I4J3KL $((ugXc*2C2C.D-EF  eU\\ * *""$$  " "5 : :(2(:(:(F*$$O$$ GELL--.hs7|nE  &&zz%  #((U[[8I7J(SWS`S`SbRc/de e'     rc URbUUc[RSS5$URup4UupVXV4X44:a"[RSSUSUSUSU35$URR 5HAupxUR U5n [ XyXR5n U R(aM?U s $ [R5$)a1Validate all parameters for a function call. Args: constraints: Function constraints to validate against kwargs: Keyword arguments passed to the function compute_capability: Current device's compute capability (major, minor) Returns: ValidationResult indicating success or first failure __hardware__z2CUDA compute capability required but not availablezcompute capability .z < required ) rwrrrorgetrr_rr) constraintskwargscompute_capability min_major min_minor curr_major curr_minor param_nameparam_constraintrXresults rrrs))5  %#(( T  +AA !3  #y&< <#((%j\:,l9+UVW`Vab  )4(:(:(@(@(B$  :& 3CE`E`a~~~M )C     r) rr(rXz!torch.Tensor | torch.dtype | Nonerrr_rkr%rr@)rr rz/Mapping[str, torch.Tensor | torch.dtype | None]rrvr%r) __future__rabcrr dataclassesrrtypingrrUcollections.abcr __all__rr r r rr rrrrrrrsK"#( ' $     $ =) = = $ (i ( ( $ ' ' '" $DDDD $   4  Z Z  Z +! +! ,+! +!$ +!  +!b26"!$"! ;"!/"! "!r