apply black formatter to eole/trainer.py

eole-nlp · Oct 23, 2024 · b733b73 · b733b73
1 parent 1250387
commit b733b73
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eole/trainer.py b/eole/trainer.py
@@ -14,7 +14,7 @@
 import torch
 import traceback
 import eole.utils
-from eole.utils.distributed import all_gather_list,all_reduce_and_rescale_tensors
+from eole.utils.distributed import all_gather_list, all_reduce_and_rescale_tensors
 from eole.utils.loss import LossCompute
 from eole.utils.logging import logger
 from eole.utils.misc import clear_gpu_cache, get_autocast