diff --git a/dap/algos/addmask.py b/dap/algos/addmask.py index 41e8576..cdfaf3c 100644 --- a/dap/algos/addmask.py +++ b/dap/algos/addmask.py @@ -1,6 +1,6 @@ #TODO: find a better way to handle this -def calc_apply_additional_mask(detector, pixel_mask_pf): +def calc_apply_additional_mask(results, detector, pixel_mask_pf): apply_additional_mask = results.get("apply_additional_mask", False) if not apply_additional_mask: return diff --git a/dap/worker.py b/dap/worker.py index 07fe014..4c701a8 100644 --- a/dap/worker.py +++ b/dap/worker.py @@ -160,7 +160,7 @@ def work(backend_address, accumulator_host, accumulator_port, visualisation_host pixel_mask_pf = None - calc_apply_additional_mask(detector, pixel_mask_pf) + calc_apply_additional_mask(results, detector, pixel_mask_pf) if pixel_mask_corrected is not None: