diff --git a/dap/algos/mask.py b/dap/algos/mask.py index 7c73ef1..febd40f 100644 --- a/dap/algos/mask.py +++ b/dap/algos/mask.py @@ -5,7 +5,7 @@ def calc_mask_pixels(data, pixel_mask_pf): if pixel_mask_pf is None: return - data[pixel_mask_pf != 1] = np.nan #TODO: boolean mask + data[pixel_mask_pf == 0] = np.nan #TODO: boolean mask diff --git a/dap/algos/radprof.py b/dap/algos/radprof.py index b1b3fcf..5a958e0 100644 --- a/dap/algos/radprof.py +++ b/dap/algos/radprof.py @@ -3,7 +3,7 @@ import numpy as np def calc_radial_integration(results, data, keep_pixels, pixel_mask_pf, center_radial_integration, r_radial_integration): if keep_pixels is None and pixel_mask_pf is not None: - keep_pixels = (pixel_mask_pf != 0) #TODO: boolean mask + keep_pixels = (pixel_mask_pf == 1) #TODO: boolean mask if center_radial_integration is None: center_radial_integration = [