fix hdf5_to_dict

lanl · Dec 23, 2024 · eb2b689 · kelslund · Dec 23, 2024 · eb2b689
1 parent 29cf5ef
commit eb2b689
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/script/analysis/hdf5_to_dict.py b/script/analysis/hdf5_to_dict.py
@@ -402,7 +402,8 @@ def load_dump(fname, geom=None, nulegacy=False):
       Ns = dump['local_Ns'][:]
       wb2N = Ns*wmean*wmean
       w2 = dump['local_wsqr'][:]
-      dump['local_stddev'] = np.sqrt(wb2N + N2*(w2 - wb2N)/(Ns - 1))
+      ones = np.ones_like(Ns)
+      dump['local_stddev'] = np.sqrt(wb2N + Ns*(w2 - wb2N)/(np.minimum(ones, Ns - 1))
       if hdr['NEUTRINO_OSCILLATIONS']:
         dump['local_osc_rate'] = dfile['local_osc_count'][:]