diff --git a/examples/port_datasets/slurm_mirror_dataset.py b/examples/port_datasets/slurm_mirror_dataset.py index a05e14fa8..c348a260c 100644 --- a/examples/port_datasets/slurm_mirror_dataset.py +++ b/examples/port_datasets/slurm_mirror_dataset.py @@ -81,6 +81,7 @@ class MirrorVideos(PipelineStep): self.vcodec = vcodec def run(self, data=None, rank: int = 0, world_size: int = 1): + import logging import subprocess from pathlib import Path @@ -92,6 +93,7 @@ class MirrorVideos(PipelineStep): init_logging() disable_progress_bars() + logger = logging.getLogger(__name__) def swap_left_right_name(name: str) -> str: result = name.replace("left_", "LEFT_PLACEHOLDER_") @@ -186,6 +188,7 @@ class MirrorDataAndMetadata(PipelineStep): if rank != 0: return + import logging from pathlib import Path import numpy as np @@ -199,6 +202,7 @@ class MirrorDataAndMetadata(PipelineStep): init_logging() disable_progress_bars() + logger = logging.getLogger(__name__) MIRRORING_MASK = { "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_frames": dataset.meta.total_frames, "total_tasks": dataset.meta.total_tasks, - "total_videos": dataset.meta.total_videos, "total_chunks": dataset.meta.total_chunks, }) write_info(new_meta.info, new_meta.root)