Source code for statsmodels.iolib.smpickle

"""Helper files for pickling"""
from statsmodels.iolib.openfile import get_file_obj


[docs]def save_pickle(obj, fname): """ Save the object to file via pickling. Parameters ---------- fname : {str, pathlib.Path} Filename to pickle to """ import pickle with get_file_obj(fname, "wb") as fout: pickle.dump(obj, fout, protocol=-1)
[docs]def load_pickle(fname): """ Load a previously saved object .. 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, pathlib.Path} Filename to unpickle Notes ----- This method can be used to load *both* models and results. """ import pickle with get_file_obj(fname, "rb") as fin: return pickle.load(fin)