SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix audio feature extractor deps (#24636)

Browse Source 
* Fix audio feature extractor deps

* use audio utils window over torch window
This commit is contained in:
Sanchit Gandhi
2023-07-04 16:03:27 +01:00
committed by GitHub
parent cd4584e3c8
commit 4e94566018
11 changed files with 28 additions and 119 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/models/speecht5/test_feature_extraction_speecht5.py
View File

@@ -20,16 +20,13 @@ import unittest
import numpy as np
from transformers import BatchFeature, is_speech_available
from transformers import BatchFeature, SpeechT5FeatureExtractor
from transformers.testing_utils import require_torch
from transformers.utils.import_utils import is_torch_available
from ...test_sequence_feature_extraction_common import SequenceFeatureExtractionTestMixin
if is_speech_available():
from transformers import SpeechT5FeatureExtractor
if is_torch_available():
import torch
@@ -142,7 +139,7 @@ class SpeechT5FeatureExtractionTester(unittest.TestCase):
@require_torch
class SpeechT5FeatureExtractionTest(SequenceFeatureExtractionTestMixin, unittest.TestCase):
feature_extraction_class = SpeechT5FeatureExtractor if is_speech_available() else None
feature_extraction_class = SpeechT5FeatureExtractor
def setUp(self):
self.feat_extract_tester = SpeechT5FeatureExtractionTester(self)