Fix the flavors order in Yoshid model loader

SNEWS2 · May 23, 2024 · 1e5b69a · 1e5b69a
1 parent 417c686
commit 1e5b69a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/snewpy/models/presn_loaders.py b/python/snewpy/models/presn_loaders.py
@@ -158,6 +158,10 @@ def __init__(self, filename):
                 np.linspace(10,20,501)[1:]
             ])
         dNdEdT = np.stack(data, axis=1)
+        #rearrange flavors from NU_E, NU_E_BAR,_NU_X, NU_X_BAR to current 
+        indices = np.argsort([Flavor.NU_E,Flavor.NU_E_BAR, Flavor.NU_X, Flavor.NU_X_BAR]) 
+        dNdEdT = dNdEdT.take(indices, axis=0)
+
         self.interpolated = _interp_TE(
             times, energies, dNdEdT, ax_t=1, ax_e=2
         )