Update mask correctly based on results

equinor · Nov 12, 2018 · 15f4e18 · 15f4e18
1 parent 4395d14
commit 15f4e18
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/python/python/ert_gui/simulation/models/ensemble_smoother.py b/python/python/ert_gui/simulation/models/ensemble_smoother.py
@@ -82,9 +82,10 @@ def create_context(self, arguments, prior_context = None):
             mask = arguments["active_realizations"]
         else:
             itr = 1
-            mask = prior_context.get_mask( )
             sim_fs = prior_context.get_target_fs( )
             target_fs = None
+            state = RealizationStateEnum.STATE_HAS_DATA | RealizationStateEnum.STATE_INITIALIZED
+            mask = sim_fs.getStateMap().createMask(state)
 
         run_context = ErtRunContext.ensemble_smoother( sim_fs, target_fs, mask, runpath_fmt, jobname_fmt, subst_list, itr)
         return run_context

diff --git a/python/python/ert_gui/simulation/models/multiple_data_assimilation.py b/python/python/ert_gui/simulation/models/multiple_data_assimilation.py
@@ -173,7 +173,8 @@ def create_context(self, arguments, itr, prior_context = None, update = True):
         if prior_context is None:
             mask = arguments["active_realizations"]
         else:
-            mask = prior_context.get_mask( )
+            state = RealizationStateEnum.STATE_HAS_DATA | RealizationStateEnum.STATE_INITIALIZED
+            mask = sim_fs.getStateMap().createMask(state)
 
         run_context = ErtRunContext.ensemble_smoother( sim_fs, target_fs, mask, runpath_fmt, jobname_fmt, subst_list, itr)
         return run_context