fix(dataset): key is an action if it starts with such prefix in dataset_to_policy_features (#1156)

2025-05-28 15:08:10 +02:00
parent f6198d20c6
commit 0db1a67eaf
1 changed files with 1 additions and 1 deletions
--- a/lerobot/common/datasets/utils.py
+++ b/lerobot/common/datasets/utils.py
@@ -461,7 +461,7 @@ def dataset_to_policy_features(features: dict[str, dict]) -> dict[str, PolicyFea
            type = FeatureType.ENV
        elif key.startswith("observation"):
            type = FeatureType.STATE
-        elif key == "action":
+        elif key.startswith("action"):
            type = FeatureType.ACTION
        else:
            continue