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

Add # Copied from statements to audio feature extractors that use the floats_list function (#26581)

Browse Source 
Add # Copied from statements to audio feature extractors that use the floats_list function.
This commit is contained in:
dg845
2023-10-04 08:09:48 -07:00
committed by GitHub
parent 0a49f909bc
commit 9deb18ca1a
8 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/models/wav2vec2/test_feature_extraction_wav2vec2.py
View File

@@ -29,6 +29,7 @@ from ...test_sequence_feature_extraction_common import SequenceFeatureExtraction
global_rng = random.Random()
# Copied from tests.models.whisper.test_feature_extraction_whisper.floats_list
def floats_list(shape, scale=1.0, rng=None, name=None):
"""Creates a random float32 tensor"""
if rng is None:

1
tests/models/wav2vec2/test_tokenization_wav2vec2.py
View File

@@ -39,6 +39,7 @@ from ...test_tokenization_common import TokenizerTesterMixin
global_rng = random.Random()
# Copied from tests.models.whisper.test_feature_extraction_whisper.floats_list
def floats_list(shape, scale=1.0, rng=None, name=None):
"""Creates a random float32 tensor"""
if rng is None: