admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 03:59:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

import logging

Browse Source
This commit is contained in:
Pepijn
2026-02-05 16:15:53 +01:00
parent 76a4529d29
commit aceb651e40
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/port_datasets/slurm_mirror_dataset.py
+4 -1
View File
@@ -81,6 +81,7 @@ class MirrorVideos(PipelineStep):
self.vcodec = vcodec self.vcodec = vcodec
def run(self, data=None, rank: int = 0, world_size: int = 1): def run(self, data=None, rank: int = 0, world_size: int = 1):
import logging
import subprocess import subprocess
from pathlib import Path from pathlib import Path
@@ -92,6 +93,7 @@ class MirrorVideos(PipelineStep):
init_logging() init_logging()
disable_progress_bars() disable_progress_bars()
logger = logging.getLogger(__name__)
def swap_left_right_name(name: str) -> str: def swap_left_right_name(name: str) -> str:
result = name.replace("left_", "LEFT_PLACEHOLDER_") result = name.replace("left_", "LEFT_PLACEHOLDER_")
@@ -186,6 +188,7 @@ class MirrorDataAndMetadata(PipelineStep):
if rank != 0: if rank != 0:
return return
import logging
from pathlib import Path from pathlib import Path
import numpy as np import numpy as np
@@ -199,6 +202,7 @@ class MirrorDataAndMetadata(PipelineStep):
init_logging() init_logging()
disable_progress_bars() disable_progress_bars()
logger = logging.getLogger(__name__)
MIRRORING_MASK = { MIRRORING_MASK = {
"joint_1": -1, "joint_2": -1, "joint_3": -1, "joint_4": 1, "joint_1": -1, "joint_2": -1, "joint_3": -1, "joint_4": 1,
@@ -338,7 +342,6 @@ class MirrorDataAndMetadata(PipelineStep):
"total_episodes": dataset.meta.total_episodes, "total_episodes": dataset.meta.total_episodes,
"total_frames": dataset.meta.total_frames, "total_frames": dataset.meta.total_frames,
"total_tasks": dataset.meta.total_tasks, "total_tasks": dataset.meta.total_tasks,
"total_videos": dataset.meta.total_videos,
"total_chunks": dataset.meta.total_chunks, "total_chunks": dataset.meta.total_chunks,
}) })
write_info(new_meta.info, new_meta.root) write_info(new_meta.info, new_meta.root)