Store error sd in obs vec

StevePny · Oct 4, 2024 · 5151378 · 5151378
1 parent d7db346
commit 5151378
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dabench/observer/_observer.py b/dabench/observer/_observer.py
@@ -323,8 +323,10 @@ def observe(self):
                                     size=errors_vec_size)
 
         # Include flag for whether observations are stationary or not
+        # Also include error_sd as an attribute
         obs_vec = obs_vec.assign_attrs(
-            stationary_observers=self.stationary_observers)
+            stationary_observers=self.stationary_observers,
+            error_sd=error_sd)
 
         # Clip errors to positive only
         if self.error_positive_only: