From b49e4016f2ddec891c0f14b32899e8734d851d4c Mon Sep 17 00:00:00 2001
From: Steven Palma <steven.palma@huggingface.co>
Date: Mon, 20 Apr 2026 16:17:23 +0200
Subject: [PATCH] test(rollout): fix expectations

---
 src/lerobot/rollout/__init__.py | 1 -
 tests/test_rollout.py           | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/lerobot/rollout/__init__.py b/src/lerobot/rollout/__init__.py
index 8ef6c26cd..da8c18acc 100644
--- a/src/lerobot/rollout/__init__.py
+++ b/src/lerobot/rollout/__init__.py
@@ -56,7 +56,6 @@ __all__ = [
     "DAggerPedalConfig",
     "DAggerStrategyConfig",
     "DatasetContext",
-    "DatasetRecordConfig",
     "HardwareContext",
     "HighlightStrategyConfig",
     "InferenceEngine",
diff --git a/tests/test_rollout.py b/tests/test_rollout.py
index 0a53ea477..dd3fb25b6 100644
--- a/tests/test_rollout.py
+++ b/tests/test_rollout.py
@@ -80,7 +80,7 @@ def test_dagger_config_defaults():
     from lerobot.rollout import DAggerStrategyConfig
 
     cfg = DAggerStrategyConfig()
-    assert cfg.num_episodes == 10
+    assert cfg.num_episodes is None
     assert cfg.record_autonomous is False
     assert cfg.input_device == "keyboard"