From e7d04737dd3732c7c545147fddce6ed025bfa5a3 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Tue, 23 Dec 2025 20:06:36 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 unsloth/models/rl_replacements.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/unsloth/models/rl_replacements.py b/unsloth/models/rl_replacements.py
index 8436ca0dc9..f0f0386bd1 100644
--- a/unsloth/models/rl_replacements.py
+++ b/unsloth/models/rl_replacements.py
@@ -393,7 +393,14 @@ def grpo_trainer__generate_and_score_completions(function_name, function):
 
     if match:
         indent = match.group(1)
-        new_code = indent + "if not _was_training:\n" + indent + "    self.model.for_inference()\n" + indent + "return output"
+        new_code = (
+            indent
+            + "if not _was_training:\n"
+            + indent
+            + "    self.model.for_inference()\n"
+            + indent
+            + "return output"
+        )
         function = function.replace(f"{indent}return output", new_code)
 
     return function