From c83785b74f58d8026ec4cbb3c90f3cf21dd30db7 Mon Sep 17 00:00:00 2001
From: Blunde1 <lundeberent@gmail.com>
Date: Tue, 21 May 2024 22:48:34 +0200
Subject: [PATCH] store statistical objects

---
 src/ert/analysis/_es_update.py | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git a/src/ert/analysis/_es_update.py b/src/ert/analysis/_es_update.py
index 7b590a8dba9..46f77276d55 100644
--- a/src/ert/analysis/_es_update.py
+++ b/src/ert/analysis/_es_update.py
@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import logging
+import pickle
 import time
 from collections import defaultdict
 from datetime import datetime
@@ -568,7 +569,9 @@ def adaptive_localization_progress_callback(
                 cov_shrinkage=True,
                 symmetrization=True,
                 shrinkage_target=2,
-                inflation_factor=np.log(graph_u_sub.number_of_edges()),
+                inflation_factor=np.log(
+                    graph_u_sub.number_of_nodes() + graph_u_sub.number_of_edges()
+                ),
             )
 
             # Add to block-diagonal full precision
@@ -582,6 +585,31 @@ def adaptive_localization_progress_callback(
             format="csc",
         )
 
+        with open("../output/01_drogon_ahm_no_seismic/X_full.pkl", "wb") as file:
+            pickle.dump(X_full, file)
+
+        with open("../output/01_drogon_ahm_no_seismic/prec_u.pkl", "wb") as file:
+            pickle.dump(Prec_u, file)
+
+        with open("../output/01_drogon_ahm_no_seismic/S.pkl", "wb") as file:
+            pickle.dump(S, file)
+
+        with open(
+            "../output/01_drogon_ahm_no_seismic/observation_values.pkl", "wb"
+        ) as file:
+            pickle.dump(observation_values, file)
+
+        with open("../output/01_drogon_ahm_no_seismic/Prec_eps.pkl", "wb") as file:
+            pickle.dump(Prec_eps, file)
+
+        with open(
+            "../output/01_drogon_ahm_no_seismic/observation_errors.pkl", "wb"
+        ) as file:
+            pickle.dump(observation_errors, file)
+
+        with open("../output/01_drogon_ahm_no_seismic/H.pkl", "wb") as file:
+            pickle.dump(H, file)
+
         # Initialize EnIF object with full precision matrices
         eps = 1e-3  # for better condition number
         gtmap = EnIF(