fix comment

alibaba · Dec 27, 2024 · 7c75953 · 7c75953
1 parent 7e3fcd0
commit 7c75953
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chatlearn/utils/dist_utils.py b/chatlearn/utils/dist_utils.py
@@ -27,7 +27,7 @@ def bucket_tensor_generator(tensor_generator, bucket_size_mb):
         tensor_generator (Generator): A generator of tensors to be separated into chunks.
         size_limit (int): The limit of each chunk in bytes.
 
-    Return:
+    Yield:
         dense_buckets: Blocks of tensors of same type and within size_limit.
         sparse_bucket: A list of sparse tensors
     """