Modify observation_features

2026-05-14 16:19:45 +00:00 · 2025-07-28 17:59:14 +02:00
parent 2d646fbc64
commit 32b08a242d
1 changed files with 9 additions and 5 deletions
@@ -78,11 +78,15 @@ class Reachy2Robot(Robot):

    @property
    def observation_features(self) -> dict:
-        return {
-            "dtype": "float32",
-            "shape": len(REACHY2_MOTORS),
-            "names": {"motors": list(REACHY2_MOTORS)},
-        }
+        return dict.fromkeys(
+            REACHY2_MOTORS.keys(),
+            float,
+        )
+        # return {
+        #     "dtype": "float32",
+        #     "shape": len(REACHY2_MOTORS),
+        #     "names": {"motors": list(REACHY2_MOTORS)},
+        # }

    @property
    def action_features(self) -> dict: