From 062cc6055cc718c441a7cce03d2197882aa6b17e Mon Sep 17 00:00:00 2001 From: Sven Augustin Date: Fri, 2 Aug 2024 15:10:52 +0200 Subject: [PATCH] use the same validation logic as in other cases --- dap/algos/mask.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/dap/algos/mask.py b/dap/algos/mask.py index 1ed79fa..0015d49 100644 --- a/dap/algos/mask.py +++ b/dap/algos/mask.py @@ -2,8 +2,10 @@ import numpy as np def calc_mask_pixels(pfdata, pixel_mask_pf): - if pixel_mask_pf is not None: - pfdata[pixel_mask_pf != 1] = np.nan #TODO: boolean mask + if pixel_mask_pf is None: + return + + pfdata[pixel_mask_pf != 1] = np.nan #TODO: boolean mask