Ensure correct array type with ES

Important to make sure fields that should be float32 never get cast to float64, which would waste memory.
equinor · Mar 21, 2024 · 64fe249 · 64fe249
1 parent a71a6df
commit 64fe249
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/ert/analysis/_es_update.py b/src/ert/analysis/_es_update.py
@@ -524,11 +524,9 @@ def adaptive_localization_progress_callback(
             )
 
         else:
-            # The batch of parameters
-            X_local = temp_storage[param_group]
-
-            # Update manually using global transition matrix T
-            temp_storage[param_group] = X_local @ T
+            temp_storage[param_group] = temp_storage[param_group] @ T.astype(
+                temp_storage[param_group].dtype
+            )
 
         log_msg = f"Storing data for {param_group}.."
         _logger.info(log_msg)