diff --git a/bin/pluviometer/site_filter.py b/bin/pluviometer/site_filter.py index b8e3ee8..18a02f4 100644 --- a/bin/pluviometer/site_filter.py +++ b/bin/pluviometer/site_filter.py @@ -12,9 +12,9 @@ def apply(self, variant_data: RNASiteVariantData) -> None: if variant_data.coverage >= self.cov_threshold: np.copyto( self.frequencies, - variant_data.frequencies * variant_data.frequencies - >= self.edit_threshold, + variant_data.frequencies ) + self.frequencies[self.frequencies < self.edit_threshold] = 0 else: self.frequencies.fill(0)