Merge pull request #237 from SNEWS2/Sheshuk/fix_preSN

Transpose the preSN flux arrays in , to match the (Ntime,Nenergy) shape, same as in CCSN
SNEWS2 · May 16, 2023 · 08c3b84 · 08c3b84
2 parents 9096246 + 4360a6e
commit 08c3b84
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/snewpy/models/presn.py b/python/snewpy/models/presn.py
@@ -67,7 +67,7 @@ def get_initial_spectra(self, t, E, flavors=Flavor):
         alpha = np.expand_dims(alpha, 0)
         b = np.expand_dims(b, 0)
         fluence = a * Enu ** alpha * np.exp(-b * Enu) / (u.MeV * u.s)
-        result = {f: fluence * self.factor[f] for f in flavors}
+        result = {f: fluence.T * self.factor[f] for f in flavors}
         return result
 
 
@@ -103,7 +103,7 @@ def get_initial_spectra(self, t, E, flavors=Flavor):
         t = np.array(-t.to_value("hour"), ndmin=1)
         E = np.array(E.to_value("MeV"), ndmin=1)
         flux = self.interpolated(t, E) / (u.MeV * u.s)
-        return {f: flux[f].T for f in flavors}
+        return {f: flux[f] for f in flavors}
 
 class Kato_2017(SupernovaModel):
     """Set up a presupernova model based on 
@@ -138,5 +138,5 @@ def get_initial_spectra(self, t, E, flavors=Flavor):
         t = np.array(-t.to_value("s"), ndmin=1)
         E = np.array(E.to_value("MeV"), ndmin=1)
         flux = self.interpolated(t, E) / (u.MeV * u.s)
-        return {f: flux[f].T for f in flavors}
+        return {f: flux[f] for f in flavors}