Merge branch 'nl_fix' of https://github.com/apax-hub/apax into nl_fix

apax-hub · Apr 5, 2024 · 0a9ec1c · 0a9ec1c
2 parents 2305f50 + 1dc1203
commit 0a9ec1c
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/apax/train/trainer.py b/apax/train/trainer.py
@@ -130,12 +130,10 @@ def fit(
             epoch_loss["val_loss"] /= val_steps_per_epoch
             epoch_loss["val_loss"] = float(epoch_loss["val_loss"])
 
-            epoch_metrics.update(
-                {
-                    f"val_{key}": float(val)
-                    for key, val in val_batch_metrics.compute().items()
-                }
-            )
+            epoch_metrics.update({
+                f"val_{key}": float(val)
+                for key, val in val_batch_metrics.compute().items()
+            })
 
         epoch_metrics.update({**epoch_loss})