Skip to content

Commit 3e2751c

Browse files
authored
Save state as dictionary (#138)
1 parent f64c204 commit 3e2751c

1 file changed

Lines changed: 2 additions & 4 deletions

File tree

pipt/loop/assimilation.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -341,17 +341,15 @@ def _save_analysis_debug(self):
341341
else:
342342
analysisdebug = [self.ensemble.keys_da['analysisdebug']]
343343

344-
if 'state' in analysisdebug:
345-
analysisdebug.remove('state')
346-
analysisdebug.append('enX')
347-
348344
# Loop over variables to store in save list
349345
for save_typ in analysisdebug:
350346
if hasattr(self, save_typ):
351347
save_dict[save_typ] = eval('self.{}'.format(save_typ))
352348
elif hasattr(self.ensemble, save_typ):
353349
save_dict[save_typ] = eval('self.ensemble.{}'.format(save_typ))
354350
# Save with key equal variable name and the actual variable
351+
elif save_typ == 'state':
352+
save_dict['state'] = entools.matrix_to_dict(self.ensemble.enX, self.ensemble.idX)
355353
else:
356354
print(f'Cannot save {save_typ}, because it is a local variable!\n\n')
357355

0 commit comments

Comments
 (0)