[Refactor]: fix linting errors to pass pre-commit checks

Author: Ibinarriaga8

sota-implementations/cql/utils.py

@@ -18,8 +18,7 @@
     TensorDictPrioritizedReplayBuffer,
     TensorDictReplayBuffer,
 )
-from torchrl.data.datasets.d4rl import D4RLExperienceReplay
-from torchrl.data.datasets.minari_data  import MinariExperienceReplay 
+from torchrl.data.datasets.minari_data import MinariExperienceReplay
 from torchrl.data.replay_buffers import SamplerWithoutReplacement
 from torchrl.envs import (
     CatTensors,

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 = []