From 010cca7fe4cd6fb0a3b59ad0a6a494ffc5c4ad9c Mon Sep 17 00:00:00 2001 From: Yoach Lacombe Date: Thu, 12 Sep 2024 17:33:17 +0200 Subject: [PATCH] make fixup --- tests/models/whisper/test_modeling_whisper.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/models/whisper/test_modeling_whisper.py b/tests/models/whisper/test_modeling_whisper.py index 5d46bdcb8bb9a1..c677e40011e0fb 100644 --- a/tests/models/whisper/test_modeling_whisper.py +++ b/tests/models/whisper/test_modeling_whisper.py @@ -2385,7 +2385,9 @@ def test_tiny_token_timestamp_generation_longform(self): ) inputs = inputs.to(torch_device) - generate_outputs = model.generate(**inputs, return_segments=True, return_token_timestamps=True, return_timestamps=True) + generate_outputs = model.generate( + **inputs, return_segments=True, return_token_timestamps=True, return_timestamps=True + ) token_timestamps_shape = [ [segment["token_timestamps"].shape for segment in segment_list]