diff --git a/infer/modules/train/train.py b/infer/modules/train/train.py
index b03d572..b81a0d2 100644
--- a/infer/modules/train/train.py
+++ b/infer/modules/train/train.py
@@ -116,7 +116,11 @@ def main():
         children[i].join()
 
 
-def run(rank, n_gpus, hps,):
+def run(
+    rank,
+    n_gpus,
+    hps,
+):
     global global_step
     if rank == 0:
         logger = utils.get_logger(hps.model_dir)