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/encodec/test_feature_extraction_encodec.py
View File

@@ -20,16 +20,13 @@ import unittest
import numpy as np
from transformers import is_speech_available
from transformers import EncodecFeatureExtractor
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 EncodecFeatureExtractor
if is_torch_available():
import torch
@@ -103,7 +100,7 @@ class EnCodecFeatureExtractionTester(unittest.TestCase):
@require_torch
class EnCodecFeatureExtractionTest(SequenceFeatureExtractionTestMixin, unittest.TestCase):
feature_extraction_class = EncodecFeatureExtractor if is_speech_available() else None
feature_extraction_class = EncodecFeatureExtractor
def setUp(self):
self.feat_extract_tester = EnCodecFeatureExtractionTester(self)