mirror of
https://github.com/huggingface/lerobot.git
synced 2026-08-08 17:39:44 +00:00
feat(data): add recipe-driven language supervision (#4182)
* feat(data): add recipe-driven language supervision * test(collate): expect preserved language columns * Address PR review feedback * Address Claude review feedback
This commit is contained in:
@@ -38,6 +38,13 @@ def test_dataset_config_empty_episodes_ok():
|
||||
DatasetConfig(repo_id="user/repo", episodes=[])
|
||||
|
||||
|
||||
def test_dataset_config_ignores_negative_excluded_episodes(caplog):
|
||||
config = DatasetConfig(repo_id="user/repo", exclude_episodes=[-2, 1, -1, 3])
|
||||
|
||||
assert config.exclude_episodes == [1, 3]
|
||||
assert "Ignoring negative exclude_episodes entries: [-2, -1]" in caplog.text
|
||||
|
||||
|
||||
def test_dataset_config_bucket_streaming_ok():
|
||||
DatasetConfig(repo_id="user/repo", repo_type="bucket", streaming=True)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user