Only perturb observations once

No need to perturb per parameter batch
equinor · Dec 14, 2023 · 174f548 · 174f548
1 parent f3063c4
commit 174f548
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/ert/analysis/_es_update.py b/src/ert/analysis/_es_update.py
@@ -545,6 +545,10 @@ def analysis_ES(
             # Pre-calculate cov_YY
             cov_YY = np.cov(S)
 
+            D = smoother_adaptive_es.perturb_observations(
+                ensemble_size=ensemble_size, alpha=1.0
+            )
+
         else:
             # Compute transition matrix so that
             # X_posterior = X_prior @ T
@@ -576,9 +580,6 @@ def analysis_ES(
 
                 for param_batch_idx in TimedIterator(batches, progress_callback):
                     X_local = temp_storage[param_group.name][param_batch_idx, :]
-                    D = smoother_adaptive_es.perturb_observations(
-                        ensemble_size=ensemble_size, alpha=1.0
-                    )
                     temp_storage[param_group.name][
                         param_batch_idx, :
                     ] = smoother_adaptive_es.assimilate(