mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-22 12:09:42 +00:00
fix(test): linting issue
This commit is contained in:
@@ -224,7 +224,7 @@ def test_complex_nested_observation():
|
|||||||
|
|
||||||
# Check that all observation keys are preserved
|
# Check that all observation keys are preserved
|
||||||
original_obs_keys = {k for k in batch if k.startswith("observation.")}
|
original_obs_keys = {k for k in batch if k.startswith("observation.")}
|
||||||
reconstructed_obs_keys = {k for k in reconstructed_batch.keys() if k.startswith("observation.")}
|
reconstructed_obs_keys = {k for k in reconstructed_batch if k.startswith("observation.")}
|
||||||
|
|
||||||
assert original_obs_keys == reconstructed_obs_keys
|
assert original_obs_keys == reconstructed_obs_keys
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user