M/Phi"dZddlZddZddZdS)z4 SARIMAX tools. Author: Chad Fulton License: BSD-3 Nc tj|}|dnd|z}tj||tkst d|d|d|t}tj|dkrt d|z|jd kr|jd d kr |dddf}n!|jd krt d|d |d |jdkr| }n t|dkrd}nd|v}tj |d kd k}tj|d k}|s|r6|rt d |ztj |d kdd z}ntj |}t|dkrd}nXtj|tjd t|d zkr |d}n|}t!|t"o)tjtj|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 [/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/tsa/arima/tools.pystandardize_lag_orderr" sL HUOOE}GG*u*FGGG zQ5;q>Q..aaad  aj7z"validate_basic..s"EEEjG,,EEEr#z,Parameters vector%s includes invalid values.z/Parameters vector%s includes NaN or Inf values.rszOSpecification%s implies %d parameter%s, but values with shape %s were provided.)r r objectravelrr)float TypeErrorrisnanisinf atleast_1dsqueezer)paramslength allow_infnanr is_complexr&plurals r!validate_basicr;os}6-BBY%6E"&///EEfllnnEEE z??5&... """G !"" "" "RVBHV$4$455"VBHV$4$455"J !"" "]2:f-- . .F|y  {{@!666<@ABB B Ms A,A88B)N)FN)__doc__numpyr r"r;r(r#r!r>sT bbbbJ666666r#