We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 17a211b commit 4b0f3d3Copy full SHA for 4b0f3d3
tests/unittests/audio/test_stoi.py
@@ -20,6 +20,7 @@
20
from torch import Tensor
21
from torchmetrics.audio import ShortTimeObjectiveIntelligibility
22
from torchmetrics.functional.audio import short_time_objective_intelligibility
23
+from torchmetrics.utilities.imports import _TORCH_GREATER_EQUAL_2_0
24
25
from unittests import _Input
26
from unittests._helpers import seed_all
@@ -120,6 +121,7 @@ def test_error_on_different_shape(metric_class=ShortTimeObjectiveIntelligibility
120
121
metric(torch.randn(100), torch.randn(50))
122
123
124
+@pytest.mark.skipif(not _TORCH_GREATER_EQUAL_2_0, reason="precision issue with older torch")
125
def test_on_real_audio():
126
"""Test that metric works on real audio signal."""
127
rate, ref = wavfile.read(_SAMPLE_AUDIO_SPEECH)
0 commit comments