add default for state (max_state_dim)

This commit is contained in:
Pepijn
2025-09-23 19:32:38 +02:00
parent 7f1a5c78ac
commit 0be09c4080

View File

@@ -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: