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

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2025-12-23 20:06:36 +00:00
commit 7c1d528c00

View file

@ -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