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

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

Browse Source
This commit is contained in:
Steven Palma
2025-05-28 15:08:10 +02:00
committed by GitHub
parent f6198d20c6
commit 0db1a67eaf
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/datasets/utils.py
View File

@@ -461,7 +461,7 @@ def dataset_to_policy_features(features: dict[str, dict]) -> dict[str, PolicyFea
type = FeatureType.ENV type = FeatureType.ENV
elif key.startswith("observation"): elif key.startswith("observation"):
type = FeatureType.STATE type = FeatureType.STATE
elif key == "action": elif key.startswith("action"):
type = FeatureType.ACTION type = FeatureType.ACTION
else: else:
continue continue