o װi @s@ddlZddlZddlZddZddZd ddZd d d ZdS) Nc Cszt|WStyA}z/zt|}Wn tyYnw||kr5tjd|tdd|WYd}~Std||d}~ww)NzbIn future, this will raise TypeError, as {} will need to be an integer not just an integral float.) stacklevelz{} must be an integer)operatorindex TypeErrorintwarningswarnformatDeprecationWarning)xdesceixrM/home/ubuntu/.local/lib/python3.10/site-packages/cupy/polynomial/polyutils.py_deprecate_as_ints&   rcCs6|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` rb)trimN)sizecupy trim_zeros)seqretrrrtrimseqs   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 emptyrCoefficient array is not 1-drz&Coefficient arrays have no common typecsg|] }|jddqS)Fcopy)astype).0adtyperr Lszas_series..) r ValueErrorndimr$kindravelrappendr common_type)alistrarraysr"rrr#r as_series2s      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-negativerrrrzbool inputs are not allowedFrN)r&rr'r$r(r)r rr+abs _manipulation add_remove_first_nonzero_krnlitem zeros_like)ctolfiltindrrrtrimcoefPs(      r:)T)r)rrrrrr.r:rrrrs