Update __init__.py

ensure that noisereduce can be loaded without torch
timsainb · Oct 19, 2023 · 33fcfbe · 33fcfbe
1 parent 13c9aac
commit 33fcfbe
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/noisereduce/spectralgate/__init__.py b/noisereduce/spectralgate/__init__.py
@@ -1,3 +1,9 @@
 from .nonstationary import SpectralGateNonStationary
 from .stationary import SpectralGateStationary
-from .streamed_torch_gate import StreamedTorchGate
+try:
+    import torch
+    TORCH_AVAILABLE = True
+except ImportError:
+    TORCH_AVAILABLE = False
+if TORCH_AVAILABLE:
+    from .streamed_torch_gate import StreamedTorchGate