From 9e8cc81d1dce02c2285e0eab930871471942cc3f Mon Sep 17 00:00:00 2001 From: rolo Date: Tue, 10 Mar 2026 13:09:16 +0100 Subject: [PATCH] Save state as dictionary --- pipt/loop/assimilation.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/pipt/loop/assimilation.py b/pipt/loop/assimilation.py index 8326ba9..76f26fb 100644 --- a/pipt/loop/assimilation.py +++ b/pipt/loop/assimilation.py @@ -341,10 +341,6 @@ def _save_analysis_debug(self): else: analysisdebug = [self.ensemble.keys_da['analysisdebug']] - if 'state' in analysisdebug: - analysisdebug.remove('state') - analysisdebug.append('enX') - # Loop over variables to store in save list for save_typ in analysisdebug: if hasattr(self, save_typ): @@ -352,6 +348,8 @@ def _save_analysis_debug(self): elif hasattr(self.ensemble, save_typ): save_dict[save_typ] = eval('self.ensemble.{}'.format(save_typ)) # Save with key equal variable name and the actual variable + elif save_typ == 'state': + save_dict['state'] = entools.matrix_to_dict(self.ensemble.enX, self.ensemble.idX) else: print(f'Cannot save {save_typ}, because it is a local variable!\n\n')