a ;e@sLddlmZddlmZmZmZddlmZddlmZm Z m Z m Z ddl m Z mZmZmZe rlddlmZddlZdd d d d d d dddZdd d d d d d dddZdd d d d d d dddZddd dd d dd d dddZddd dd d dd d dddZddd dd d dd d dd d!Zddd dd d dd d dd"d#ZdS)$) annotations)_real_floating_dtypes_real_numeric_dtypes_numeric_dtypes)Array)float32float64 complex64 complex128) TYPE_CHECKINGOptionalTupleUnion)DtypeNFaxiskeepdimsrz%Optional[Union[int, Tuple[int, ...]]]bool)xrrreturncCs*|jtvrtdttj|j||dS)Nz+Only real numeric dtypes are allowed in maxr)dtyper TypeErrorr_newnpmax_arrayrrrrZ/var/www/sunrise/env/lib/python3.9/site-packages/numpy/array_api/_statistical_functions.pyrs rcCs*|jtvrtdttj|j||dS)Nz3Only real floating-point dtypes are allowed in meanr)rrrrrrmeanrrrrrr s r cCs*|jtvrtdttj|j||dS)Nz+Only real numeric dtypes are allowed in minr)rrrrrrminrrrrrr!+s r!rrrzOptional[Dtype])rrrrrcCsR|jtvrtd|dur8|jtkr*t}n|jtkr8t}tt j |j |||dS)Nz'Only numeric dtypes are allowed in prod)rrr) rrrrr r r rrrprodrrrrrrrrr#7s   r#g)r correctionrzUnion[int, float])rrr%rrcCs,|jtvrtdttj|j|||dS)Nz2Only real floating-point dtypes are allowed in stdrZddofr)rrrrrrstdrrrr%rrrrr'Ls r'cCsR|jtvrtd|dur8|jtkr*t}n|jtkr8t}tt j |j |||dS)Nz&Only numeric dtypes are allowed in sumr") rrrrr r r rrrsumrr$rrrr)Zs   r)cCs,|jtvrtdttj|j|||dS)Nz2Only real floating-point dtypes are allowed in varr&)rrrrrrvarrr(rrrr*os r*) __future__rZ_dtypesrrrZ _array_objectrrr r r typingr r rr_typingrnumpyrrr r!r#r'r)r*rrrrs@