admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 02:06:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(processor): make RenderMessagesStep.recipe optional for non-recipe pipelines

Browse Source
This commit is contained in:
Khalil Meftah
2026-07-10 14:05:08 +02:00
parent f5c2ee1753
commit c043a6c418
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/processor/render_messages_processor.py
+4 -1
View File
@@ -40,11 +40,14 @@ class RenderMessagesStep(ProcessorStep):
``message_streams`` / ``target_message_indices`` keys. ``message_streams`` / ``target_message_indices`` keys.
""" """
recipe: TrainingRecipe recipe: TrainingRecipe | None = None
dataset_ctx: Any | None = None dataset_ctx: Any | None = None
def __call__(self, transition: EnvTransition) -> EnvTransition | None: def __call__(self, transition: EnvTransition) -> EnvTransition | None:
"""Render messages for a single transition; return ``None`` to drop it.""" """Render messages for a single transition; return ``None`` to drop it."""
if self.recipe is None:
return transition
complementary_data = transition.get(TransitionKey.COMPLEMENTARY_DATA) or {} complementary_data = transition.get(TransitionKey.COMPLEMENTARY_DATA) or {}
persistent = complementary_data.get(LANGUAGE_PERSISTENT) or [] persistent = complementary_data.get(LANGUAGE_PERSISTENT) or []
events = complementary_data.get(LANGUAGE_EVENTS) or [] events = complementary_data.get(LANGUAGE_EVENTS) or []