From dd562d28a45a25c9f1e98a7c9b51a76449e5dba0 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Wed, 28 May 2025 16:06:50 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 colossalai/checkpoint_io/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/colossalai/checkpoint_io/utils.py b/colossalai/checkpoint_io/utils.py
index 0fbd65d86..ea97da1ba 100644
--- a/colossalai/checkpoint_io/utils.py
+++ b/colossalai/checkpoint_io/utils.py
@@ -1142,7 +1142,7 @@ def gather_state_dict_fast(
                 ks.append(k)
                 ops.append(dist.P2POp(dist.irecv, buffer, dist.get_global_rank(group, i), group))
         reqs = dist.batch_isend_irecv(ops)
-        for req in reqs: # len(reqs) maybe be different from len(ops) because of coalescing
+        for req in reqs:  # len(reqs) maybe be different from len(ops) because of coalescing
             req.wait()
         for k in ks:
             returned_state_dict[k] = returned_state_dict[k].to(device)