admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-23 09:46:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(eval): use FeatureType enum comparison instead of string value

Browse Source
This commit is contained in:
Khalil Meftah
2026-06-15 18:50:24 +02:00
parent 58cf6c8710
commit e069557228
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/scripts/lerobot_eval.py
+2 -2
View File
@@ -72,7 +72,7 @@ from termcolor import colored
from torch import Tensor, nn from torch import Tensor, nn
from tqdm import trange from tqdm import trange
from lerobot.configs import parser from lerobot.configs import FeatureType, parser
from lerobot.configs.eval import EvalPipelineConfig from lerobot.configs.eval import EvalPipelineConfig
from lerobot.datasets.lerobot_dataset import LeRobotDataset from lerobot.datasets.lerobot_dataset import LeRobotDataset
from lerobot.envs import ( from lerobot.envs import (
@@ -104,7 +104,7 @@ def _env_features_to_dataset_features(env_features: dict, raw_obs: dict | None =
""" """
features = {} features = {}
for key, ft in env_features.items(): for key, ft in env_features.items():
if ft.type.value == "visual": if ft.type is FeatureType.VISUAL:
shape = tuple(ft.shape) shape = tuple(ft.shape)
if raw_obs is not None and key in raw_obs and isinstance(raw_obs[key], np.ndarray): if raw_obs is not None and key in raw_obs and isinstance(raw_obs[key], np.ndarray):
shape = raw_obs[key].shape[1:] # strip batch dim shape = raw_obs[key].shape[1:] # strip batch dim