mirror of
https://github.com/huggingface/lerobot.git
synced 2026-06-03 04:11:24 +00:00
add default for state (max_state_dim)
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user