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/audio_spectrogram_transformer/test_feature_extraction_audio_spectrogram_transformer.py
View File

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

1
tests/models/encodec/test_feature_extraction_encodec.py
View File

@@ -34,6 +34,7 @@ if is_torch_available():
global_rng = random.Random() 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): def floats_list(shape, scale=1.0, rng=None, name=None):
"""Creates a random float32 tensor""" """Creates a random float32 tensor"""
if rng is None: if rng is None:

1
tests/models/musicgen/test_processing_musicgen.py
View File

@@ -35,6 +35,7 @@ if is_speech_available():
global_rng = random.Random() 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): def floats_list(shape, scale=1.0, rng=None, name=None):
"""Creates a random float32 tensor""" """Creates a random float32 tensor"""
if rng is None: if rng is None:

1
tests/models/speech_to_text/test_feature_extraction_speech_to_text.py
View File

@@ -32,6 +32,7 @@ if is_speech_available():
global_rng = random.Random() 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): def floats_list(shape, scale=1.0, rng=None, name=None):
"""Creates a random float32 tensor""" """Creates a random float32 tensor"""
if rng is None: if rng is None:

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

@@ -34,6 +34,7 @@ if is_torch_available():
global_rng = random.Random() 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): def floats_list(shape, scale=1.0, rng=None, name=None):
"""Creates a random float32 tensor""" """Creates a random float32 tensor"""
if rng is None: if rng is None:

1
tests/models/tvlt/test_feature_extraction_tvlt.py
View File

@@ -38,6 +38,7 @@ if is_datasets_available():
global_rng = random.Random() 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): def floats_list(shape, scale=1.0, rng=None, name=None):
"""Creates a random float32 tensor""" """Creates a random float32 tensor"""
if rng is None: if rng is None:

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() 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): def floats_list(shape, scale=1.0, rng=None, name=None):
"""Creates a random float32 tensor""" """Creates a random float32 tensor"""
if rng is None: 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() 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): def floats_list(shape, scale=1.0, rng=None, name=None):
"""Creates a random float32 tensor""" """Creates a random float32 tensor"""
if rng is None: if rng is None: