try

lerobot/common/datasets/video_utils.py

@@ -105,6 +105,9 @@ def decode_video_frames_torchvision(
         if current_ts >= last_ts:
             break
 
+    reader.container.close()
+    reader = None
+
     query_ts = torch.tensor(timestamps)
     loaded_ts = torch.tensor(loaded_ts)
 

tests/test_datasets.py

@@ -91,7 +91,6 @@ def test_factory(env_name, repo_id, policy_name):
             assert key in item, f"{key}"
 
 
-@pytest.mark.skip(reason="TODO(rcadene): Test is super slow during CI, but locally it runs fast")
 def test_compute_stats_on_xarm():
     """Check that the statistics are computed correctly according to the stats_patterns property.