File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 11"""Tests for forced alignment functions."""
22
3- import numpy as np
4- import pandas as pd
53import pytest
64import torch
75from transformers import Wav2Vec2ForCTC , Wav2Vec2Processor
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" )
403392def test_get_prediction_matrix (dummy_model : tuple ) -> None :
You can’t perform that action at this time.
0 commit comments