fix bug when setting the filtering none

fgcz · Aug 26, 2024 · 333019d · 333019d
1 parent 5f7df41
commit 333019d
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/src/depiction/spectrum/peak_picking/findmf_peak_picker.py b/src/depiction/spectrum/peak_picking/findmf_peak_picker.py
@@ -37,9 +37,12 @@ def pick_peaks(
             area=self.area,
             max_peaks=self.max_peaks,
         )
-        return self.peak_filtering.filter_peaks(
-            spectrum_mz_arr=mz_arr,
-            spectrum_int_arr=int_arr,
-            peak_mz_arr=peak_mz_arr,
-            peak_int_arr=peak_int_arr,
-        )
+        if self.peak_filtering is not None:
+            return self.peak_filtering.filter_peaks(
+                spectrum_mz_arr=mz_arr,
+                spectrum_int_arr=int_arr,
+                peak_mz_arr=peak_mz_arr,
+                peak_int_arr=peak_int_arr,
+            )
+        else:
+            return peak_mz_arr, peak_int_arr