run black

flairNLP · Jul 17, 2023 · 2057f71 · 2057f71
1 parent de8293b
commit 2057f71
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/flair/trainers/trainer.py b/flair/trainers/trainer.py
@@ -591,7 +591,9 @@ def train_custom(
                         total_train_samples += batch_train_samples
                         train_loss = batch_train_loss / batch_train_samples
                         self._record(MetricRecord.scalar(("train", "batch_loss"), train_loss, total_train_samples))
-                        self._record(MetricRecord.scalar(("train", "gradient_norm"), gradient_norm, total_train_samples))
+                        self._record(
+                            MetricRecord.scalar(("train", "gradient_norm"), gradient_norm, total_train_samples)
+                        )
 
                         epoch_train_loss += batch_train_loss
                         epoch_train_samples += batch_train_samples