fix on_magnitudes

Author: Yicun Duan

case_studies/redshift/redshift_from_img/encoder/metrics.py

@@ -80,7 +80,7 @@ def update(self, true_cat, est_cat, matching):
                 true_red = true_cat["redshifts"][i, tcat_matches, :].to(self.device)
                 est_red = est_cat["redshifts"][i, ecat_matches, :].to(self.device)
 
-            true_mag = true_cat.on_magnitudes(zero_point=3631)[i][..., self.mag_band][
+            true_mag = true_cat.on_magnitudes(zero_point=3631e9)[i][..., self.mag_band][
                 tcat_matches
             ].to(self.device)
             bin_indices = torch.bucketize(true_mag, cutoffs)
@@ -254,7 +254,7 @@ def update(self, true_cat, est_cat, matching):
                 est_red = est_cat["redshifts"][i, ecat_matches, :].to(self.device)
 
             if self.bin_type == "ab_mag":
-                true_mag = true_cat.on_magnitudes(zero_point=3631)[i][..., self.mag_band][
+                true_mag = true_cat.on_magnitudes(zero_point=3631e9)[i][..., self.mag_band][
                     tcat_matches
                 ].to(self.device)
             elif self.bin_type == "njy":
@@ -343,7 +343,7 @@ def update(self, true_cat, est_cat, matching):
                 est_red = est_cat["redshifts"][i, ecat_matches, :].to(self.device)
 
             if self.bin_type == "ab_mag":
-                true_mag = true_cat.on_magnitudes(zero_point=3631)[i][..., self.mag_band][
+                true_mag = true_cat.on_magnitudes(zero_point=3631e9)[i][..., self.mag_band][
                     tcat_matches
                 ].to(self.device)
             elif self.bin_type == "njy":
@@ -425,7 +425,7 @@ def update(self, true_cat, est_cat, matching):
                 est_red = est_cat["redshifts"][i, ecat_matches, :].to(self.device)
 
             if self.bin_type == "ab_mag":
-                true_mag = true_cat.on_magnitudes(zero_point=3631)[i][..., self.mag_band][
+                true_mag = true_cat.on_magnitudes(zero_point=3631e9)[i][..., self.mag_band][
                     tcat_matches
                 ].to(self.device)
             elif self.bin_type == "njy":
@@ -523,7 +523,7 @@ def update(self, true_cat, est_cat, matching):
                 est_red = est_cat["redshifts"][i, ecat_matches, :].to(self.device)
 
             if self.bin_type == "ab_mag":
-                true_mag = true_cat.on_magnitudes(zero_point=3631)[i][..., self.mag_band][
+                true_mag = true_cat.on_magnitudes(zero_point=3631e9)[i][..., self.mag_band][
                     tcat_matches
                 ].to(self.device)
             elif self.bin_type == "njy":
@@ -613,7 +613,7 @@ def update(self, true_cat, est_cat, matching):
                 est_red = est_cat["redshifts"][i, ecat_matches, :].to(self.device)
 
             if self.bin_type == "ab_mag":
-                true_mag = true_cat.on_magnitudes(zero_point=3631)[i][..., self.mag_band][
+                true_mag = true_cat.on_magnitudes(zero_point=3631e9)[i][..., self.mag_band][
                     tcat_matches
                 ].to(self.device)
             elif self.bin_type == "njy":