mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-22 12:09:42 +00:00
test(rollout): fix expectations
This commit is contained in:
@@ -56,7 +56,6 @@ __all__ = [
|
|||||||
"DAggerPedalConfig",
|
"DAggerPedalConfig",
|
||||||
"DAggerStrategyConfig",
|
"DAggerStrategyConfig",
|
||||||
"DatasetContext",
|
"DatasetContext",
|
||||||
"DatasetRecordConfig",
|
|
||||||
"HardwareContext",
|
"HardwareContext",
|
||||||
"HighlightStrategyConfig",
|
"HighlightStrategyConfig",
|
||||||
"InferenceEngine",
|
"InferenceEngine",
|
||||||
|
|||||||
@@ -80,7 +80,7 @@ def test_dagger_config_defaults():
|
|||||||
from lerobot.rollout import DAggerStrategyConfig
|
from lerobot.rollout import DAggerStrategyConfig
|
||||||
|
|
||||||
cfg = DAggerStrategyConfig()
|
cfg = DAggerStrategyConfig()
|
||||||
assert cfg.num_episodes == 10
|
assert cfg.num_episodes is None
|
||||||
assert cfg.record_autonomous is False
|
assert cfg.record_autonomous is False
|
||||||
assert cfg.input_device == "keyboard"
|
assert cfg.input_device == "keyboard"
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user