From abae2288e793bda74a8bc6c33ac744ce48a305a0 Mon Sep 17 00:00:00 2001 From: Lisa Dorofeeva Date: Tue, 15 Jul 2025 08:50:35 +0200 Subject: [PATCH] Final cleanup --- dap/algos/streakfind.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/dap/algos/streakfind.py b/dap/algos/streakfind.py index 8ee069c..0c3a7f6 100644 --- a/dap/algos/streakfind.py +++ b/dap/algos/streakfind.py @@ -8,7 +8,6 @@ https://github.com/simply-nicky/streak_finder/ """ import h5py import numpy as np -from psutil import virtual_memory from streak_finder.label import Structure2D from streak_finder.streak_finder import detect_peaks, detect_streaks, filter_peaks from streak_finder._src.src.median import median @@ -52,7 +51,6 @@ def calc_streakfinder_analysis(results, data, pf_pixel_mask): _calc_streakfinder_analysis(results, snr, pf_pixel_mask) except Exception as error: # Broad exception - we don't want to break anything here results["cbd_error"] = f"StreakFind - Error processing CBD data:\n{error}" - print(f"mem used after {virtual_memory().used // pow(2, 30)} Gb \n") return data @@ -142,8 +140,8 @@ def _calc_streakfinder_analysis(results, snr, mask): ] if not all(param in results.keys() for param in params_required): - print(f"ERROR: Not enough parameters for streak finder analysis. Skipping.\n" - f"{params_required=}") + results["cbd_error"] = (f"ERROR: Not enough parameters for streak finder analysis. " + f"Skipping.\n{params_required=}") return peak_structure_radius = results["cbd_peak_structure_radius"] # peak @@ -164,7 +162,6 @@ def _calc_streakfinder_analysis(results, snr, mask): y_center = results.get("beam_center_y", None) mask_rois = results.get("cbd_mask_rois", []) # list of [x_min, x_max, y_min, y_max] - for mask_roi in mask_rois: mask[mask_roi[2]: mask_roi[3], mask_roi[0]: mask_roi[1]] = False