make fixup

ylacombe · Sep 12, 2024 · 010cca7 · 010cca7
1 parent a1217e8
commit 010cca7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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]