diff --git a/src/mednet/scripts/upload.py b/src/mednet/scripts/upload.py
index ea659e2246116b5e243632acde3585ce77e2de9d..f02d09343d0b5ecc524bfda2d3717d5e4cdc2291 100644
--- a/src/mednet/scripts/upload.py
+++ b/src/mednet/scripts/upload.py
@@ -189,7 +189,9 @@ def upload(
         click.secho(
             f"  -> `#accumulations` ({train_data['accumulate-grad-batches']})"
         )
-        mlflow.log_param("#Accumulations", train_data["accumulate-grad-batches"])
+        mlflow.log_param(
+            "#Accumulations", train_data["accumulate-grad-batches"]
+        )
         click.secho(f"  -> `epoch (best)` ({best_epoch})")
         mlflow.log_param("Epoch (best)", best_epoch)