mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-22 03:59:42 +00:00
test(processor): fix batch expectation
This commit is contained in:
@@ -1175,4 +1175,4 @@ def test_task_processing_creates_new_transition():
|
|||||||
# The task should be processed correctly (wrapped in list)
|
# The task should be processed correctly (wrapped in list)
|
||||||
assert result[TransitionKey.COMPLEMENTARY_DATA]["task"] == ["sort_objects"]
|
assert result[TransitionKey.COMPLEMENTARY_DATA]["task"] == ["sort_objects"]
|
||||||
# Original complementary data is also modified (current behavior)
|
# Original complementary data is also modified (current behavior)
|
||||||
assert original_comp_data["task"] == ["sort_objects"]
|
assert original_comp_data["task"] == "sort_objects"
|
||||||
|
|||||||
Reference in New Issue
Block a user