admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-24 03:37:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

adding legacy to existing tests

Browse Source
This commit is contained in:
Maximellerbach
2026-06-03 16:52:44 +02:00
parent 350f8e8d4d
commit a617b0eda2
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_rollout.py
+5
View File
@@ -60,6 +60,7 @@ def test_strategy_config_types():
BaseStrategyConfig,
DAggerStrategyConfig,
HighlightStrategyConfig,
LegacyStrategyConfig,
SentryStrategyConfig,
)
@@ -67,6 +68,7 @@ def test_strategy_config_types():
assert SentryStrategyConfig().type == "sentry"
assert HighlightStrategyConfig().type == "highlight"
assert DAggerStrategyConfig().type == "dagger"
assert LegacyStrategyConfig().type == "legacy"
def test_dagger_config_invalid_input_device():
@@ -203,6 +205,8 @@ def test_create_strategy_dispatches():
BaseStrategyConfig,
DAggerStrategy,
DAggerStrategyConfig,
LegacyStrategy,
LegacyStrategyConfig,
SentryStrategy,
SentryStrategyConfig,
create_strategy,
@@ -211,6 +215,7 @@ def test_create_strategy_dispatches():
assert isinstance(create_strategy(BaseStrategyConfig()), BaseStrategy)
assert isinstance(create_strategy(SentryStrategyConfig()), SentryStrategy)
assert isinstance(create_strategy(DAggerStrategyConfig()), DAggerStrategy)
assert isinstance(create_strategy(LegacyStrategyConfig()), LegacyStrategy)
def test_create_strategy_unknown_raises():