admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-15 08:39:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(processor): specialized processors respect contract by raising if none (#1909)

Browse Source 
* fix(processor): specialized processor now raise

* test(processor): fix tests for now raise specialized processors

* test(processor): use identity in newly introduced pipeline
This commit is contained in:
Steven Palma
2025-09-10 18:45:47 +02:00
committed by GitHub
parent 51588f741b
commit 6745958362
7 changed files with 78 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/processor/test_sac_processor.py
+1 -1
View File
@@ -372,7 +372,7 @@ def test_sac_processor_edge_cases():
assert processed[TransitionKey.ACTION].shape == (1, 5)
# Test with None action
transition = create_transition(observation={OBS_STATE: torch.randn(10)}, action=None)
transition = create_transition(observation={OBS_STATE: torch.randn(10)}, action={})
processed = preprocessor(transition)
assert processed[TransitionKey.OBSERVATION][OBS_STATE].shape == (1, 10)
# When action is None, it may still be present with None value