From 9deb18ca1a2266bc73c05468780b17b92e6a27e5 Mon Sep 17 00:00:00 2001 From: dg845 <58458699+dg845@users.noreply.github.com> Date: Wed, 4 Oct 2023 08:09:48 -0700 Subject: [PATCH] Add # Copied from statements to audio feature extractors that use the floats_list function (#26581) Add # Copied from statements to audio feature extractors that use the floats_list function. --- .../test_feature_extraction_audio_spectrogram_transformer.py | 1 + tests/models/encodec/test_feature_extraction_encodec.py | 1 + tests/models/musicgen/test_processing_musicgen.py | 1 + .../speech_to_text/test_feature_extraction_speech_to_text.py | 1 + tests/models/speecht5/test_feature_extraction_speecht5.py | 1 + tests/models/tvlt/test_feature_extraction_tvlt.py | 1 + tests/models/wav2vec2/test_feature_extraction_wav2vec2.py | 1 + tests/models/wav2vec2/test_tokenization_wav2vec2.py | 1 + 8 files changed, 8 insertions(+) diff --git a/tests/models/audio_spectrogram_transformer/test_feature_extraction_audio_spectrogram_transformer.py b/tests/models/audio_spectrogram_transformer/test_feature_extraction_audio_spectrogram_transformer.py index 69a1bddc82..85d696479c 100644 --- a/tests/models/audio_spectrogram_transformer/test_feature_extraction_audio_spectrogram_transformer.py +++ b/tests/models/audio_spectrogram_transformer/test_feature_extraction_audio_spectrogram_transformer.py @@ -33,6 +33,7 @@ if is_torch_available(): import torch +# 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: diff --git a/tests/models/encodec/test_feature_extraction_encodec.py b/tests/models/encodec/test_feature_extraction_encodec.py index bf40135c32..5a8010d247 100644 --- a/tests/models/encodec/test_feature_extraction_encodec.py +++ b/tests/models/encodec/test_feature_extraction_encodec.py @@ -34,6 +34,7 @@ if is_torch_available(): 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: diff --git a/tests/models/musicgen/test_processing_musicgen.py b/tests/models/musicgen/test_processing_musicgen.py index 41962552ff..c0157ed0b5 100644 --- a/tests/models/musicgen/test_processing_musicgen.py +++ b/tests/models/musicgen/test_processing_musicgen.py @@ -35,6 +35,7 @@ if is_speech_available(): 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: diff --git a/tests/models/speech_to_text/test_feature_extraction_speech_to_text.py b/tests/models/speech_to_text/test_feature_extraction_speech_to_text.py index 293b33fde8..d8929c4ef0 100644 --- a/tests/models/speech_to_text/test_feature_extraction_speech_to_text.py +++ b/tests/models/speech_to_text/test_feature_extraction_speech_to_text.py @@ -32,6 +32,7 @@ if is_speech_available(): 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: diff --git a/tests/models/speecht5/test_feature_extraction_speecht5.py b/tests/models/speecht5/test_feature_extraction_speecht5.py index a03914991e..22d99a8180 100644 --- a/tests/models/speecht5/test_feature_extraction_speecht5.py +++ b/tests/models/speecht5/test_feature_extraction_speecht5.py @@ -34,6 +34,7 @@ if is_torch_available(): 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: diff --git a/tests/models/tvlt/test_feature_extraction_tvlt.py b/tests/models/tvlt/test_feature_extraction_tvlt.py index ad37b1f984..166f31021c 100644 --- a/tests/models/tvlt/test_feature_extraction_tvlt.py +++ b/tests/models/tvlt/test_feature_extraction_tvlt.py @@ -38,6 +38,7 @@ if is_datasets_available(): 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: diff --git a/tests/models/wav2vec2/test_feature_extraction_wav2vec2.py b/tests/models/wav2vec2/test_feature_extraction_wav2vec2.py index 556f01c6b2..c4ab09b606 100644 --- a/tests/models/wav2vec2/test_feature_extraction_wav2vec2.py +++ b/tests/models/wav2vec2/test_feature_extraction_wav2vec2.py @@ -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: diff --git a/tests/models/wav2vec2/test_tokenization_wav2vec2.py b/tests/models/wav2vec2/test_tokenization_wav2vec2.py index 174d3009a9..3ab8717d81 100644 --- a/tests/models/wav2vec2/test_tokenization_wav2vec2.py +++ b/tests/models/wav2vec2/test_tokenization_wav2vec2.py @@ -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: