scatterin: add missing cp import

INGV · May 27, 2024 · 292bf1d · 292bf1d
1 parent 9488f16
commit 292bf1d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/scatcluster/processing/scattering.py b/scatcluster/processing/scattering.py
@@ -11,8 +11,7 @@
 from obspy.core.stream import Stream
 from scatseisnet.network import ScatteringNetwork
 from scatseisnet.operation import segmentize
-
-from scatcluster.helper import is_gpu_available
+import cupy as cp
 import xarray as xr
 from tqdm import tqdm