admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-16 00:59:46 +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_diffusion_processor.py
+5 -1
View File
@@ -218,7 +218,11 @@ def test_diffusion_processor_without_stats():
"""Test Diffusion processor creation without dataset statistics."""
config = create_default_config()
preprocessor, postprocessor = make_diffusion_pre_post_processors(config, dataset_stats=None)
preprocessor, postprocessor = make_diffusion_pre_post_processors(
config,
dataset_stats=None,
preprocessor_kwargs={"to_transition": identity_transition, "to_output": identity_transition},
)
# Should still create processors
assert preprocessor is not None