chore: pre-commit linting fixes

src/axolotl/utils/distributed.py

@@ -142,13 +142,15 @@ def reduce_and_broadcast(fn1, fn2):
     if not is_distributed():
         return fn2([fn1()])
 
-    gathered_values = gather_scalar_from_all_ranks(fn1, world_size=dist.get_world_size())
+    gathered_values = gather_scalar_from_all_ranks(
+        fn1, world_size=dist.get_world_size()
+    )
 
     # Use compute_and_broadcast to compute the reduced value on the main process
     # and then broadcast it to all ranks
     return compute_and_broadcast(lambda: fn2(gathered_values))
 
-  
+
 def broadcast_dict(vals: dict):
     if not is_distributed():
         return vals

src/axolotl/utils/trainer.py

@@ -34,7 +34,12 @@
 )
 from axolotl.utils.collators import DataCollatorForSeq2Seq
 from axolotl.utils.dataloader import MultipackDistributedDataloader
-from axolotl.utils.distributed import is_distributed, is_main_process, reduce_and_broadcast, zero_first
+from axolotl.utils.distributed import (
+    is_distributed,
+    is_main_process,
+    reduce_and_broadcast,
+    zero_first,
+)
 from axolotl.utils.schedulers import get_cosine_schedule_with_quadratic_warmup
 
 LOG = logging.getLogger("axolotl")