From b1ff7132c1b7c560654fdc22139cb60ca9940741 Mon Sep 17 00:00:00 2001
From: Pepijn <pepijn@huggingface.co>
Date: Sat, 30 Aug 2025 16:08:21 +0200
Subject: [PATCH] add decode logging

---
 src/lerobot/scripts/train.py | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/src/lerobot/scripts/train.py b/src/lerobot/scripts/train.py
index a034e21dd..d235770ac 100644
--- a/src/lerobot/scripts/train.py
+++ b/src/lerobot/scripts/train.py
@@ -295,13 +295,7 @@ def train(cfg: TrainPipelineConfig):
     torch.backends.cuda.matmul.allow_tf32 = True
 
     logging.info("Creating dataset")
-    # Pass video backend to dataset for RLearN optimization
-    dataset_kwargs = {}
-    if getattr(cfg.policy, "type", None) == "rlearn" and hasattr(cfg.policy, "video_backend"):
-        dataset_kwargs["video_backend"] = cfg.policy.video_backend
-        logging.info(f"Using video backend: {cfg.policy.video_backend}")
-    
-    dataset = make_dataset(cfg, **dataset_kwargs)
+    dataset = make_dataset(cfg)
     
     # Add video decoding timing for RLearN debugging
     if getattr(cfg.policy, "type", None) == "rlearn":