admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-27 06:29:47 +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_pi0_openpi.py
+3 -3
View File
@@ -19,7 +19,7 @@ def test_policy_instantiation():
print("Testing PI0OpenPI policy instantiation...")
# Create config
config = PI0OpenPIConfig(action_dim=7, state_dim=14, dtype="float32")
config = PI0OpenPIConfig(max_action_dim=7, max_state_dim=14, dtype="float32")
# Create dummy dataset stats
dataset_stats = {
@@ -75,8 +75,8 @@ def test_config_creation():
try:
config = make_policy_config(
policy_type="pi0_openpi",
action_dim=7,
state_dim=14,
max_action_dim=7,
max_state_dim=14,
)
print("✓ Config created successfully through factory")
print(f" Config type: {type(config).__name__}")