fixing DM_cosmic

FRBs · Aug 26, 2024 · ad6f37f · ad6f37f
1 parent 70631c2
commit ad6f37f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/frb/galaxies/utils.py b/frb/galaxies/utils.py
@@ -70,7 +70,7 @@ def load_specdb(specdb_file=None):
             specdb_file = specdb_files[0]
             print("Loading spectra from {:s}".format(specdb_file))
         else:
-            raise IOError("There are no FRB_specdb.hdf5 files in your SPECDB folder")
+            raise IOError("There are no FRB_specDB_*.hdf5 files in your SPECDB folder")
     # Load it up
     specDB = SpecDB(db_file=specdb_file)
     # Return

diff --git a/frb/scripts/pz_dm.py b/frb/scripts/pz_dm.py
@@ -51,8 +51,9 @@ def main(pargs):
     print("-----------------------------------------------------")
     print(f"NE2001 = {DM_ISM:.2f}")
 
-    # DM extragalactic
-    DM_extragalactic = pargs.DM_FRB - DM_ISM.value - pargs.dm_host - pargs.dm_mwhalo
+    # DM cosmic and EG
+    DM_cosmic = pargs.DM_FRB - DM_ISM.value - pargs.dm_mwhalo
+    DM_extragalactic = DM_cosmic + pargs.dm_host  
 
     # Redshift estimates
 
@@ -74,7 +75,8 @@ def main(pargs):
     PDM_z = sdict['PDM_z']
     z = sdict['z']
     DM = sdict['DM']
-    DM_cosmic = DM_extragalactic + pargs.dm_host  #DM_cosmic = DM_extragalactic + DM_host
+
+    # Grab the right entry
     iDM = np.argmin(np.abs(DM - DM_cosmic))
     PzDM = PDM_z[iDM, :] / np.sum(PDM_z[iDM, :])