[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

apax/config/common.py

@@ -32,8 +32,7 @@ def parse_config(config: Union[str, os.PathLike, dict], mode: str = "train") ->
 
 
 def flatten(dictionary, parent_key="", separator="_"):
-    """https://stackoverflow.com/questions/6027558/flatten-nested-dictionaries-compressing-keys
-    """
+    """https://stackoverflow.com/questions/6027558/flatten-nested-dictionaries-compressing-keys"""
     items = []
     for key, value in dictionary.items():
         new_key = parent_key + separator + key if parent_key else key

apax/train/trainer.py

@@ -107,12 +107,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})