U Kv f@sFddlZddlZddlmZGdddZddZddZd d ZdS) N)dedentc@sVeZdZdZiZiZddZddZddZdd Z d d Z dd dZ e ddZ dS)ResultsWrapperz Class which wraps a statsmodels estimation Results class and steps in to reattach metadata to results (if available) cCs||_|j|_dSN)_results__doc__)selfresultsr sz*ResultsWrapper.__dir__..)dirrrr r r __dir__szResultsWrapper.__dir__csfdd}z |d}Wntk r,YnXz ||WStk rLYnXt||}|jj}j|}|rt|tr|j||df|dd}n|r|j||d}|S)Ncs t|Sr)object__getattribute__)namerr r z1ResultsWrapper.__getattribute__..rr)how) AttributeErrorgetattrmodeldata _wrap_attrsget isinstancetuple wrap_output)rattrrrobjrrr rr rs"      zResultsWrapper.__getattribute__cCs|jSr)__dict__rr r r __getstate__,szResultsWrapper.__getstate__cCs|j|dSr)r$update)rZdict_r r r __setstate__0szResultsWrapper.__setstate__FcCs&ddlm}|r||||dS)a Save a pickle of this instance. Parameters ---------- fname : {str, handle} Either a filename or a valid file handle. remove_data : bool If False (default), then the instance is pickled without changes. If True, then all arrays with length nobs are set to None before pickling. See the remove_data method. In some cases not all arrays will be set to None. r) save_pickleN)statsmodels.iolib.smpickler( remove_data)rfnamer*r(r r r save4s zResultsWrapper.savecCsddlm}||S)a Load a pickled results instance .. warning:: Loading pickled models is not secure against erroneous or maliciously constructed data. Never unpickle data received from an untrusted or unauthenticated source. Parameters ---------- fname : {str, handle} A string filename or a file handle. Returns ------- Results The unpickled results instance. r) load_pickle)r)r-)clsr+r-r r r loadIs zResultsWrapper.loadN)F)__name__ __module__ __qualname__rr _wrap_methodsr rrr%r'r, classmethodr/r r r r rs rcGsi}|D]}||q|Sr)r&)Zdictsresultdr r r union_dictsbs r7csTtfdd}t}t|}|jr:t|jnd}dj||f|_|S)Ncsjt|d}|jj}rJttrJ||f||ddd}nrf||f||}|S)Nrrr)rrrrrr r!)rargskwargsrrr#funcrr r wrapperjs (zmake_wrapper..wrapperz %s%s %s) functoolswrapsinspect signaturestrrrr0)r;rr<sig formatteddocr r:r make_wrapperis rFcCsD|jD]4\}}t||sq t||}t||}t|||q dSr)r3itemshasattrrrFsetattr)klassZwrappingmethrr;r<r r r populate_wrapper}s    rL)r>r@textwraprrr7rFrLr r r r s  \