fix empty lines

alibaba · Jan 3, 2025 · 4526338 · 4526338
1 parent 6c8ebeb
commit 4526338
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/chatlearn/utils/dist_utils.py b/chatlearn/utils/dist_utils.py
@@ -18,12 +18,15 @@
 import torch
 from torch._utils import _flatten_dense_tensors, _unflatten_dense_tensors
 
+
 def bucket_tensors(tensors, bucket_size_mb):
     """Group tensors into chunks. We seperate sparse and dense tensor,
     each containing tensors of same type up to certain byte limit in total size.
+
     Args:
         tensors (Sequence): A sequence of tensors to be separated into chunks.
         size_limit (int): The limit of each chunk in bytes.
+
     Return:
         dense_buckets: Blocks of tensors of same type and within size_limit.
         sparse_bucket: A list of sparse tensors