[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-20 07:02:30 +00:00 committed by numb3r33
commit c8e7bd9f09

View file

@ -388,12 +388,12 @@ def grpo_trainer__generate_and_score_completions(function_name, function):
patched = patched[: match.start()] + wrapped + patched[match.end() :]
function = patched
function = function.replace(
" return output",
""" if not _was_training:
" return output",
""" if not _was_training:
self.model.for_inference()
return output"""
return output""",
)
return function