From b6a72ecce01315dc88428c26ac5bfa4300c4fd60 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 5 Feb 2025 05:47:23 -0800 Subject: [PATCH] Update rl.py --- unsloth/models/rl.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index 0e9e28b48c..caf12cd6d5 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -89,9 +89,9 @@ def NotebookProgressCallback_on_log(Trainer_metrics): pass -def _NotebookTrainingTracker_write_line(Trainer_metrics): +def NotebookTrainingTracker_write_line(Trainer_metrics): set_Trainer_metrics = set(Trainer_metrics) - def NotebookTrainingTracker_write_line(self, values): + def _NotebookTrainingTracker_write_line(self, values): """ Write the values in the inner table. @@ -132,7 +132,7 @@ def _NotebookTrainingTracker_write_line(Trainer_metrics): pass pass pass - return NotebookTrainingTracker_write_line + return _NotebookTrainingTracker_write_line pass