tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

style(deprecated method): remove no longer used get_features_from_robot function (replaced by hw_to_dataset_features) (#1560)

Browse Source
This commit is contained in:
Caroline Pascal
2025-07-21 19:12:03 +02:00
committed by GitHub
parent f6ec1d89a5
commit 9b9f4757fb
1 changed files with 0 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -41,7 +41,6 @@ from lerobot.datasets.backward_compatibility import (
BackwardCompatibilityError, BackwardCompatibilityError,
ForwardCompatibilityError, ForwardCompatibilityError,
) )
from lerobot.robots import Robot
from lerobot.utils.utils import is_valid_numpy_dtype_string from lerobot.utils.utils import is_valid_numpy_dtype_string
DEFAULT_CHUNK_SIZE = 1000 # Max number of episodes per chunk DEFAULT_CHUNK_SIZE = 1000 # Max number of episodes per chunk
@@ -440,16 +439,6 @@ def build_dataset_frame(
return frame return frame
def get_features_from_robot(robot: Robot, use_videos: bool = True) -> dict:
camera_ft = {}
if robot.cameras:
camera_ft = {
key: {"dtype": "video" if use_videos else "image", **ft}
for key, ft in robot.camera_features.items()
}
return {**robot.motor_features, **camera_ft, **DEFAULT_FEATURES}
def dataset_to_policy_features(features: dict[str, dict]) -> dict[str, PolicyFeature]: def dataset_to_policy_features(features: dict[str, dict]) -> dict[str, PolicyFeature]:
# TODO(aliberts): Implement "type" in dataset features and simplify this # TODO(aliberts): Implement "type" in dataset features and simplify this
policy_features = {} policy_features = {}