format black

flairNLP · Feb 2, 2024 · 9841c8b · 9841c8b
1 parent 85f4c89
commit 9841c8b
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/flair/trainers/language_model_trainer.py b/flair/trainers/language_model_trainer.py
@@ -227,9 +227,7 @@ def train(
                 optimizer.load_state_dict(self.optimizer_state)
 
             if isinstance(optimizer, (AdamW, SGDW)):
-                scheduler: ReduceLROnPlateau = ReduceLRWDOnPlateau(
-                    optimizer, factor=anneal_factor, patience=patience
-                )
+                scheduler: ReduceLROnPlateau = ReduceLRWDOnPlateau(optimizer, factor=anneal_factor, patience=patience)
             else:
                 scheduler = ReduceLROnPlateau(optimizer, factor=anneal_factor, patience=patience)