Update pyannote/audio/tasks/segmentation/speaker_diarization.py

Co-authored-by: Hervé BREDIN <[email protected]>
pyannote · Nov 25, 2024 · eef7a77 · eef7a77
1 parent 32fb725
commit eef7a77
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/pyannote/audio/tasks/segmentation/speaker_diarization.py b/pyannote/audio/tasks/segmentation/speaker_diarization.py
@@ -467,9 +467,7 @@ def training_step(self, batch, batch_idx: int):
 
         if not self.automatic_optimization:
             optimizers = self.model.optimizers()
-            optimizers = (
-                [optimizers] if not isinstance(optimizers, list) else optimizers
-            )
+            optimizers = optimizers if isinstance(optimizers, list) else [optimizers]
             for optimizer in optimizers:
                 optimizer.zero_grad()