U b2 @sLddlmZddlmZmZddlmZddlmZddlm Z m Z ddl m Z m Z mZmZe rlddlmZdd lZd d d d d dd dddZd d d d d dd dddZd d d d d dd dddZd d d dd d ddd dddZd dd dd d ddd ddd Zd d d dd d ddd dd!d"Zd dd dd d ddd dd#d$Zd S)%) annotations)_floating_dtypes_numeric_dtypes)Array)asarray)float32float64) TYPE_CHECKINGOptionalTupleUnion)DtypeNFaxiskeepdimsrz%Optional[Union[int, Tuple[int, ...]]]bool)xrrreturncCs*|jtkrtdttj|j||dS)Nz&Only numeric dtypes are allowed in maxr)dtyper TypeErrorr_newnpmax_arrayrrrrJ/tmp/pip-unpacked-wheel-cxuynqjx/numpy/array_api/_statistical_functions.pyrs rcCs*|jtkrtdttj|j||dS)Nz.Only floating-point dtypes are allowed in meanr)rrrrrrmeanrrrrrrs rcCs*|jtkrtdttj|j||dS)Nz&Only numeric dtypes are allowed in minr)rrrrrrminrrrrrr+s rrrrzOptional[Dtype])rrrrrcCsB|jtkrtd|dkr(|jtkr(t}ttj|j |||dS)Nz'Only numeric dtypes are allowed in prod)rrr) rrrrr rrrprodrrrrrrrrr!7s  r!g)r correctionrzUnion[int, float])rrr#rrcCs,|jtkrtdttj|j|||dS)Nz-Only floating-point dtypes are allowed in stdrZddofr)rrrrrrstdrrrr#rrrrr%Hs r%cCsB|jtkrtd|dkr(|jtkr(t}ttj|j |||dS)Nz&Only numeric dtypes are allowed in sumr ) rrrrr rrrsumrr"rrrr'Vs  r'cCs,|jtkrtdttj|j|||dS)Nz-Only floating-point dtypes are allowed in varr$)rrrrrrvarrr&rrrr(hs r() __future__rZ_dtypesrrZ _array_objectrZ_creation_functionsrrr typingr r r r _typingrZnumpyrrrrr!r%r'r(rrrrsB