Skip to content

Commit ceb53db

Browse files
committed
Remove interpolate nans related code
1 parent af54da9 commit ceb53db

1 file changed

Lines changed: 0 additions & 11 deletions

File tree

src/tests/audio/tasks/forced_alignment_test.py

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,5 @@
11
"""Tests for forced alignment functions."""
22

3-
import numpy as np
4-
import pandas as pd
53
import pytest
64
import torch
75
from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
@@ -30,7 +28,6 @@
3028
_align_transcription,
3129
_can_align_segment,
3230
_get_prediction_matrix,
33-
_interpolate_nans,
3431
_merge_repeats,
3532
_preprocess_segments,
3633
align_transcriptions,
@@ -390,14 +387,6 @@ def test_merge_repeats() -> None:
390387
assert len(segments) == 2
391388

392389

393-
@pytest.mark.skipif(not TORCHAUDIO_AVAILABLE, reason="torchaudio is not installed")
394-
def test_interpolate_nans() -> None:
395-
"""Test interpolation of NaN values."""
396-
series = pd.Series([0.0, np.nan, 2.0])
397-
interpolated_series = _interpolate_nans(series)
398-
assert interpolated_series.isnull().sum() == 0
399-
400-
401390
@pytest.mark.skipif(not TORCHAUDIO_AVAILABLE, reason="torchaudio is not installed")
402391
@pytest.mark.skipif(not torch.cuda.is_available(), reason="GPU is not available")
403392
def test_get_prediction_matrix(dummy_model: tuple) -> None:

0 commit comments

Comments
 (0)