"gi dZddlZddZddZy)z4 SARIMAX tools. Author: Chad Fulton License: BSD-3 Nc tj|}|dnd|z}tj||jtk(st d|d|d|jt}tj |dkrt d|z|jd k(r|jd d k(r |dddf}n!|jd kDrt d|d |d |jdk(r|j}nt|dk(rd}nd|v}tj|d k(d kD}tj |d kD}|s|r/|rt d |ztj|d k(dd z}ntj|}t|dk(rd}nNtj|tjd t|d zk(r|d}n|j}t!|t"xr+tj tj$|dk(}|rt d|z|S)a+ Standardize lag order input. Parameters ---------- order : int or array_like Maximum lag order (if integer) or iterable of specific lag orders. title : str, optional Description of the order (e.g. "autoregressive") to use in error messages. Returns ------- order : int or list of int Maximum lag order if consecutive lag orders were specified, otherwise a list of integer lag orders. Notes ----- It is ambiguous if order=[1] is meant to be a boolean list or a list of lag orders to include, but this is irrelevant because either interpretation gives the same result. Order=[0] would be ambiguous, except that 0 is not a valid lag order to include, so there is no harm in interpreting as a boolean list, in which case it is the same as order=0, which seems like reasonable behavior. Examples -------- >>> standardize_lag_order(3) 3 >>> standardize_lag_order(np.arange(1, 4)) 3 >>> standardize_lag_order([1, 3]) [1, 3] Norderz%s orderzInvalid z. Non-integer order (z) given.rz#Terms in the %s cannot be negative.zX. Must be an integer or 1-dimensional array-like object (e.g. list, ndarray, etc.). Got .zInvalid %s. Appears to be a boolean list (since it contains a 0 element and/or multiple elements) but also contains elements greater than 1 like a list of lag orders.z+Invalid %s. Cannot have duplicate elements.)nparrayallastypeint ValueErroranyndimshapeitemlensumwheresortarangetolist isinstancelistdiff)rtitle has_zeroshas_multiple_ones has_gt_one has_duplicates X/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/statsmodels/tsa/arima/tools.pystandardize_lag_orderr" sL HHUOE}G*u*FGG zzQ5;;q>Q.ad  a70E>EE)N)FN)__doc__numpyr r"r7r#r!r;s bJ6r#