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.
2023-10-04 08:09:48 -07:00
parent 0a49f909bc
commit 9deb18ca1a
8 changed files with 8 additions and 0 deletions
--- 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: