[Test] Fix failing test (#3033)

Author: vmoens

test/test_cost.py

@@ -9642,7 +9642,7 @@ def test_ppo_value_clipping(
                     KeyError,
                     match=f"clip_value is set to {loss_fn.clip_value}, but the key "
                     "state_value was not found in the input tensordict. "
-                    "Make sure that the value_key passed to PPO exists in "
+                    "Make sure that the.*passed to PPO exists in "
                     "the input tensordict.",
                 ):
                     loss = loss_fn(td)

torchrl/envs/llm/transforms/reason.py

@@ -153,7 +153,6 @@ def _step(
         Returns:
             The modified next_tensordict
         """
-        print("Reward", next_tensordict["reward"])
         # Handle batch dimensions
         if next_tensordict.batch_dims >= 1:
             ntds = []