hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

undo previous commit

Browse Source
This commit is contained in:
Pepijn
2025-05-16 13:59:17 +02:00
committed by Steven Palma
parent 95ebc27d32
commit 60cbde1a78
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/common/robots/lekiwi/lekiwi_client.py
View File

@@ -84,7 +84,7 @@ class LeKiwiClient(Robot):
return state_ft
@property
def action_features(self) -> dict:
def action_feature(self) -> dict:
action_ft = {
"arm_shoulder_pan": {"shape": (1,), "info": None, "dtype": "float32"},
"arm_shoulder_lift": {"shape": (1,), "info": None, "dtype": "float32"},
@@ -466,7 +466,7 @@ class LeKiwiClient(Robot):
common_keys = [
key
for key in action
if key in (motor.replace("arm_", "") for motor, _ in self.action_features.items())
if key in (motor.replace("arm_", "") for motor, _ in self.action_feature.items())
]
arm_actions = {"arm_" + arm_motor: action[arm_motor] for arm_motor in common_keys}