ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-03 12:21:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove files

Browse Source
This commit is contained in:
Jade Choghari
2025-09-11 14:18:09 +02:00
parent a19d7fb6bf
commit 4c2add41d7
15 changed files with 3 additions and 4334 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/lerobot/datasets/utils.py
View File

@@ -455,8 +455,7 @@ def dataset_to_policy_features(features: dict[str, dict]) -> dict[str, PolicyFea
shape = (shape[2], shape[0], shape[1])
elif key == "observation.environment_state":
type = FeatureType.ENV
# changed by jade
elif key.startswith("observation") or key.startswith("state"):
elif key.startswith("observation"):
type = FeatureType.STATE
elif key.startswith("action"):
type = FeatureType.ACTION