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

Removed references to batch encoding to be added later or in another PR

Browse Source
This commit is contained in:
Michel Aractingi
2025-07-18 16:52:47 +02:00
parent 5ec70f704e
commit ec40fc41b5
1 changed files with 30 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/record.py
-2
View File
@@ -73,7 +73,6 @@ from lerobot.configs.policies import PreTrainedConfig
from lerobot.datasets.image_writer import safe_stop_image_writer from lerobot.datasets.image_writer import safe_stop_image_writer
from lerobot.datasets.lerobot_dataset import LeRobotDataset from lerobot.datasets.lerobot_dataset import LeRobotDataset
from lerobot.datasets.utils import build_dataset_frame, hw_to_dataset_features from lerobot.datasets.utils import build_dataset_frame, hw_to_dataset_features
from lerobot.datasets.video_utils import VideoEncodingManager
from lerobot.policies.factory import make_policy from lerobot.policies.factory import make_policy
from lerobot.policies.pretrained import PreTrainedPolicy from lerobot.policies.pretrained import PreTrainedPolicy
from lerobot.robots import ( # noqa: F401 from lerobot.robots import ( # noqa: F401
@@ -335,7 +334,6 @@ def record(cfg: RecordConfig) -> LeRobotDataset:
listener, events = init_keyboard_listener() listener, events = init_keyboard_listener()
with VideoEncodingManager(dataset):
recorded_episodes = 0 recorded_episodes = 0
while recorded_episodes < cfg.dataset.num_episodes and not events["stop_recording"]: while recorded_episodes < cfg.dataset.num_episodes and not events["stop_recording"]:
log_say(f"Recording episode {dataset.num_episodes}", cfg.play_sounds) log_say(f"Recording episode {dataset.num_episodes}", cfg.play_sounds)