need a safety switch. Should be fixed at some point

Python-Ensemble-Toolbox · Jan 1, 2024 · 4996063 · 4996063
1 parent ae7ef8b
commit 4996063
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pipt/loop/ensemble.py b/pipt/loop/ensemble.py
@@ -39,8 +39,10 @@ def __init__(self, keys_da, keys_en, sim):
                          f'{keys_da["analysis"]} update scheme with {keys_da["energy"]} Energy.')
 
         # Internalize PIPT dictionary
-        self.keys_da = keys_da
-        self.keys_en = keys_en
+        if not hasattr(self, 'keys_da'):
+            self.keys_da = keys_da
+        if not hasattr(self, 'keys_en'):
+            self.keys_en = keys_en
 
         if self.restart is False:
             # Init in _init_prediction_output (used in run_prediction)