df: Simplify ta import

grazder · Nov 10, 2023 · 846c5f4 · 846c5f4
1 parent 78b3d14
commit 846c5f4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/DeepFilterNet/df/io.py b/DeepFilterNet/df/io.py
@@ -5,15 +5,14 @@
 import torchaudio as ta
 from loguru import logger
 from numpy import ndarray
-from packaging import version
 from torch import Tensor
 
-if version.parse(ta.__version__) >= version.parse("2.0"):
+try:
     from torchaudio import AudioMetaData
 
     TA_RESAMPLE_SINC = "sinc_interp_hann"
     TA_RESAMPLE_KAISER = "sinc_interp_kaiser"
-else:
+except ImportError:
     from torchaudio.backend.common import AudioMetaData
 
     TA_RESAMPLE_SINC = "sinc_interpolation"