skip

Tencent · Nov 22, 2024 · 14dec19 · 14dec19
1 parent fdb1a25
commit 14dec19
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 0 deletions.
diff --git a/tools/pnnx/tests/ncnn/test_torchaudio_F_inverse_spectrogram.py b/tools/pnnx/tests/ncnn/test_torchaudio_F_inverse_spectrogram.py
@@ -16,6 +16,7 @@
 import torch.nn as nn
 import torch.nn.functional as F
 import torchaudio
+from packaging import version
 
 class Model(nn.Module):
     def __init__(self):
@@ -33,6 +34,9 @@ def forward(self, x, y, z, w):
         return out0, out1, out2, out3
 
 def test():
+    if version.parse(torchaudio.__version__) < version.parse('0.10.0'):
+        return True
+
     net = Model()
     net.eval()
 

diff --git a/tools/pnnx/tests/ncnn/test_torchaudio_InverseSpectrogram.py b/tools/pnnx/tests/ncnn/test_torchaudio_InverseSpectrogram.py
@@ -16,6 +16,7 @@
 import torch.nn as nn
 import torch.nn.functional as F
 import torchaudio
+from packaging import version
 
 class Model(nn.Module):
     def __init__(self):
@@ -38,6 +39,9 @@ def forward(self, x, y, z, w):
         return out0, out1, out2, out3
 
 def test():
+    if version.parse(torchaudio.__version__) < version.parse('0.10.0'):
+        return True
+
     net = Model()
     net.eval()
 

diff --git a/tools/pnnx/tests/test_torchaudio_F_inverse_spectrogram.py b/tools/pnnx/tests/test_torchaudio_F_inverse_spectrogram.py
@@ -16,6 +16,7 @@
 import torch.nn as nn
 import torch.nn.functional as F
 import torchaudio
+from packaging import version
 
 class Model(nn.Module):
     def __init__(self):
@@ -29,6 +30,9 @@ def forward(self, x, y, z, w):
         return out0, out1, out2, out3
 
 def test():
+    if version.parse(torchaudio.__version__) < version.parse('0.10.0'):
+        return True
+
     net = Model()
     net.eval()
 

diff --git a/tools/pnnx/tests/test_torchaudio_InverseSpectrogram.py b/tools/pnnx/tests/test_torchaudio_InverseSpectrogram.py
@@ -16,6 +16,7 @@
 import torch.nn as nn
 import torch.nn.functional as F
 import torchaudio
+from packaging import version
 
 class Model(nn.Module):
     def __init__(self):
@@ -34,6 +35,9 @@ def forward(self, x, y, z, w):
         return out0, out1, out2, out3
 
 def test():
+    if version.parse(torchaudio.__version__) < version.parse('0.10.0'):
+        return True
+
     net = Model()
     net.eval()