o X۷i# @sDddlmZddlZddlZddZddZd dd Zd d d ZdS)) annotationsNc Cs:zt|WSty}z t|d||d}~ww)aA Like `operator.index`, but emits a custom exception when passed an incorrect type Parameters ---------- x : int-like Value to interpret as an integer desc : str description to include in any error message Raises ------ TypeError : if x is a float or non-numeric z must be an integer, received N)operatorindex TypeError)xdescer O/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupy/polynomial/polyutils.py_as_ints  r cCsZ|jdkr td|jdkrtd|jdkr|Stj|dd}|jdkr'|S|ddS)a9Removes small polynomial series coefficients. Args: seq (cupy.ndarray): input array. Returns: cupy.ndarray: input array with trailing zeros removed. If the resulting output is empty, it returns the first element. .. seealso:: :func:`numpy.polynomial.polyutils.trimseq` rzInput must be 1-d arrayb)trimN)ndimr ValueErrorsizecupy trim_zeros)seqretr r r trimseqs     rTcsg}|D]-}|jdkrtd|jdkrtd|jjdkr"td|}|r,t|}||qtj |fdd|D}|S) a;Returns argument as a list of 1-d arrays. Args: alist (cupy.ndarray or list of cupy.ndarray): 1-D or 2-D input array. trim (bool, optional): trim trailing zeros. Returns: list of cupy.ndarray: list of 1-D arrays. .. seealso:: :func:`numpy.polynomial.polyutils.as_series` rCoefficient array is emptyr Coefficient array is not 1-dr z&Coefficient arrays have no common typecsg|] }|jddqS)Fcopy)astype).0adtyper r Qszas_series..) rrrrkindravelrappendr common_type)alistrarraysrrr rr as_series7s      r'cCs|dkrtd|jdkrtd|jdkrtd|jjdkr$td|jdkr-|}|jt|dd }t ||kd d d }|jtj j ||j }|dkr^t|d dS|d |S) aRemoves small trailing coefficients from a polynomial. Args: c(cupy.ndarray): 1d array of coefficients from lowest to highest order. tol(number, optional): trailing coefficients whose absolute value are less than or equal to ``tol`` are trimmed. Returns: cupy.ndarray: trimmed 1d array. .. seealso:: :func:`numpy.polynomial.polyutils.trimcoef` rztol must be non-negativerr rr zbool inputs are not allowedFrN)rrrrr!r"rrr$abs _manipulation add_remove_first_nonzero_krnlitem zeros_like)ctolfiltindr r r trimcoefUs(      r3)T)r) __future__rrrr rr'r3r r r r s