fixes asyncio

pyrocko · Nov 5, 2023 · 8128f59 · 8128f59
1 parent 65beec5
commit 8128f59
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lassie/images/phase_net.py b/lassie/images/phase_net.py
@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import asyncio
 import logging
 from datetime import datetime, timedelta
 from typing import TYPE_CHECKING, Any, Literal
@@ -162,7 +163,8 @@ async def process_traces(self, traces: list[Trace]) -> list[PhaseNetImage]:
             for tr in stream:
                 tr.stats.sampling_rate = tr.stats.sampling_rate / scale
 
-        annotations: Stream = await self._phase_net.annotate_async(
+        annotations: Stream = await asyncio.to_thread(
+            self._phase_net.annotate,
             stream,
             overlap=self.window_overlap_samples,
             batch_size=self.batch_size