From 0be09c4080ffbb5ac4cf5bc9df68a9aca1c812d5 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Tue, 23 Sep 2025 19:32:38 +0200 Subject: [PATCH] add default for state (max_state_dim) --- src/lerobot/policies/pi05/processor_pi05openpi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/policies/pi05/processor_pi05openpi.py b/src/lerobot/policies/pi05/processor_pi05openpi.py index e06ae5027..545546e06 100644 --- a/src/lerobot/policies/pi05/processor_pi05openpi.py +++ b/src/lerobot/policies/pi05/processor_pi05openpi.py @@ -32,7 +32,7 @@ class Pi05PrepareStateTokenizerProcessorStep(ProcessorStep): Processor step to prepare the state and tokenize the language input. """ - max_state_dim: int + max_state_dim: int = 32 task_key: str = "task" def __call__(self, transition: EnvTransition) -> EnvTransition: