From d433f4e534c8b248adfc24846f7b6acd8f8b3145 Mon Sep 17 00:00:00 2001 From: hposborn Date: Wed, 10 Jul 2024 16:42:55 +0200 Subject: [PATCH] Fixing bug in save_model_to_file --- MonoTools/fit.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/MonoTools/fit.py b/MonoTools/fit.py index 4311c13..51ac80c 100755 --- a/MonoTools/fit.py +++ b/MonoTools/fit.py @@ -182,7 +182,7 @@ def load_model_from_file(self, loadfile=None): for key in pick: setattr(self,key,pick[key]) del pick - setattr(self,'trace',az.InferenceData.from_netcdf(loadfile.replace('_model.pickle','_trace.nc'))) + setattr(self,'trace',az.InferenceData.from_netcdf(loadfile.replace('_model.pickle','_trace.nc')).stack()) elif os.path.exists(loadfile): #Loading old version using pickle from pickled dictionary pick=pickle.load(open(loadfile,'rb')) @@ -216,7 +216,7 @@ def save_model_to_file(self, savefile=None, limit_size=False): except: print("Still a save error after unstacking") excl_types=[az.InferenceData] - cloudpickle.dumps({attr:getattr(self,attr) for attr in self.__dict__},open(savefile,'wb')) + cloudpickle.dump({attr:getattr(self,attr) for attr in self.__dict__ if type(getattr(self,attr)) not in excl_types},open(savefile,'wb')) # #Loading from pickled dictionary # saving={}