source: branches/UQ/pickle.py @ 855

#!/usr/bin/env python
#
# Author: Mike McKerns (mmckerns @caltech and @uqfoundation)
# Copyright (c) 2009-2016 California Institute of Technology.
# License: 3-clause BSD.  The full license text is available at:
#  - http://mmckerns.github.io/project/mystic/browser/mystic/LICENSE

def tmp_pickle(func, suffix='.pik'):
    """ standard pickle.dump of function to a NamedTemporaryFile """
    import dill as pickle
    import tempfile
    file = tempfile.NamedTemporaryFile(suffix=suffix, dir='.')
    pickle.dump(func, file)
    file.flush()
    return file


def unpickle(filename):
  """ standard pickle.load of function from a File """
  import dill as pickle
  return pickle.load(open(filename,'r'))