admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-24 21:19:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

rename action_dim, state_dim to max_action_dim, max_state_dim

Browse Source
This commit is contained in:
Pepijn
2025-09-17 16:34:07 +02:00
parent 8c0cdb00a6
commit 53577f5f1a
4 changed files with 29 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/test_pi05_openpi.py
+6 -6
View File
@@ -21,8 +21,8 @@ def test_pi05_model_architecture():
# Create config
config = PI05OpenPIConfig(
action_dim=7,
state_dim=14,
max_action_dim=7,
max_state_dim=14,
dtype="float32",
)
@@ -89,8 +89,8 @@ def test_pi05_forward_pass():
# Create config
config = PI05OpenPIConfig(
action_dim=7,
state_dim=14,
max_action_dim=7,
max_state_dim=14,
dtype="float32",
chunk_size=16, # Shorter chunk_size for testing
n_action_steps=16, # Shorter action steps for testing
@@ -152,8 +152,8 @@ def test_pi0_vs_pi05_differences():
print("\nComparing PI0 vs PI0.5 architectures...")
# Create both configurations
config_pi0 = PI0OpenPIConfig(action_dim=7, state_dim=14, dtype="float32")
config_pi05 = PI05OpenPIConfig(action_dim=7, state_dim=14, dtype="float32")
config_pi0 = PI0OpenPIConfig(max_action_dim=7, max_state_dim=14, dtype="float32")
config_pi05 = PI05OpenPIConfig(max_action_dim=7, max_state_dim=14, dtype="float32")
dataset_stats = {
"observation.state": {"mean": torch.zeros(14), "std": torch.ones(14)},