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")