tools/saveload.py

   1 import pickle
   2
   3 def save_obj(obj, name ):
   4     with open('obj/'+ name + '.pkl', 'wb') as f:
   5         pickle.dump(obj, f, pickle.HIGHEST_PROTOCOL)
   6
   7 def load_obj(name ):
   8     with open('obj/' + name + '.pkl', 'r') as f:
   9         return pickle.load(f)
  10
  11
  12 def load_dict(name):
  13         try:
  14                 return load_obj(name)
  15         except:
  16                 return {}