[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
2025-09-22 18:09:06 +00:00 · 2025-08-06 06:16:37 +00:00
parent 32b2148670
commit 08a1244ef1
2 changed files with 2 additions and 2 deletions
--- a/applications/ColossalChat/coati/distributed/grpo_consumer.py
+++ b/applications/ColossalChat/coati/distributed/grpo_consumer.py
@@ -530,4 +530,4 @@ class GRPOConsumer(BaseConsumer):
        model = self.policy_model.unwrap()
        state_dict = model.state_dict()
        state_dict["consumer_global_step"] = torch.tensor([self.global_step], device=self.device)
-        return state_dict
+        return state_dict