Merge 5970e0b into dd1b49c

igmhub · Feb 22, 2022 · 2bf45c1 · 2bf45c1
2 parents dd1b49c + 5970e0b
commit 2bf45c1
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/py/picca/delta_extraction/masks/dla_mask.py b/py/picca/delta_extraction/masks/dla_mask.py
@@ -146,7 +146,7 @@ def apply_mask(self, forest):
 
         # load DLAs
         if self.los_ids.get(forest.los_id) is not None:
-            dla_transmission = np.ones(len(forest.log_lambda))
+            dla_transmission = np.ones(len(lambda_))
             for (z_abs, nhi) in self.los_ids.get(forest.los_id):
                 dla_transmission *= DlaProfile(lambda_, z_abs, nhi).transmission
 
@@ -155,10 +155,10 @@ def apply_mask(self, forest):
             if len(self.mask) > 0:
                 for mask_range in self.mask:
                     for (z_abs, nhi) in self.los_ids.get(forest.los_id):
-                        w &= ((forest.log_lambda - np.log10(1. + z_abs) <
-                               mask_range['log_wave_min']) |
-                              (forest.log_lambda - np.log10(1. + z_abs) >
-                               mask_range['log_wave_max']))
+                        w &= ((lambda_ / (1. + z_abs) <
+                               10**mask_range['log_wave_min']) |
+                              (lambda_ / (1. + z_abs) >
+                               10**mask_range['log_wave_max']))
 
             # do the actual masking
             forest.transmission_correction *= dla_transmission