rescaling: bugfix

pyrocko · Nov 18, 2024 · dd4f935 · dd4f935
1 parent 3927c87
commit dd4f935
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/qseek/images/seisbench.py b/src/qseek/images/seisbench.py
@@ -293,7 +293,7 @@ def _detection_half_width(self) -> float:
     @alog_call
     async def process_traces(self, traces: list[Trace]) -> list[PhaseNetImage]:
         stream = Stream(tr.to_obspy_trace() for tr in traces)
-        if self.rescale_input != 1:
+        if self.rescale_input != 1.0:
             scale = self.rescale_input
             for tr in stream:
                 tr.stats.sampling_rate /= scale
@@ -306,7 +306,7 @@ async def process_traces(self, traces: list[Trace]) -> list[PhaseNetImage]:
             copy=False,
         )
 
-        if self.rescale_input > 1:
+        if self.rescale_input != 1.0:
             scale = self.rescale_input
             for tr in annotations:
                 tr.stats.sampling_rate *= scale