From b15b98c6975bc8e6247f54c2c76959d7fcf013b1 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 19 Feb 2025 23:15:13 -0800
Subject: [PATCH] Update rl.py

---
 unsloth/models/rl.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py
index c8602d31b2..f754fa953c 100644
--- a/unsloth/models/rl.py
+++ b/unsloth/models/rl.py
@@ -450,7 +450,7 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"):
     pass
 
     # Check if SamplingParams is in there
-    if "SamplingParams" in RLTrainer_source:
+    if "SamplingParams" in old_RLTrainer_source:
         RL_pre = RL_pre + "\n" + inspect.getsource(vLLMSamplingParams)
     pass