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

Modify observation_features

Browse Source
This commit is contained in:
glannuzel
2025-07-28 17:59:14 +02:00
parent 2d646fbc64
commit 32b08a242d
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/robots/reachy2/robot_reachy2.py
+9 -5
View File
@@ -78,11 +78,15 @@ class Reachy2Robot(Robot):
@property @property
def observation_features(self) -> dict: def observation_features(self) -> dict:
return { return dict.fromkeys(
"dtype": "float32", REACHY2_MOTORS.keys(),
"shape": len(REACHY2_MOTORS), float,
"names": {"motors": list(REACHY2_MOTORS)}, )
} # return {
# "dtype": "float32",
# "shape": len(REACHY2_MOTORS),
# "names": {"motors": list(REACHY2_MOTORS)},
# }
@property @property
def action_features(self) -> dict: def action_features(self) -> dict: