ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2025-12-23 19:20:08 +00:00 • committed by numb3r33
commit c8784ec87e
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/models/rl_replacements.py
View file

@ -388,8 +388,8 @@ def grpo_trainer__generate_and_score_completions(function_name, function):
patched = patched[: match.start()] + wrapped + patched[match.end() :]
function = patched
match = re.search(r'^(\s*)return output', function, re.MULTILINE)
match = re.search(r"^(\s*)return output", function, re.MULTILINE)
if match:
indent = match.group(1)
@ -397,7 +397,7 @@ def grpo_trainer__generate_and_score_completions(function_name, function):
f"{indent}return output",
f"""{indent}if not _was_training:
{indent} self.model.for_inference()
{indent}return output"""
{indent}return output""",
)
return function