⬆️ upgrade lerobot to 0.2.0 (#51)

robomind2lerobot/README.md

@@ -126,7 +126,7 @@ Dataset Structure of `meta/info.json`:
 
 > [!IMPORTANT]
 >
-> 1. If you want to save depth when converting the dataset, modify `_assert_type_and_shape()` function in [lerobot.common.datasets.compute_stats.py](https://github.com/huggingface/lerobot/blob/main/lerobot/common/datasets/compute_stats.py).
+> 1. If you want to save depth when converting the dataset, modify `_assert_type_and_shape()` function in [lerobot.datasets.compute_stats.py](https://github.com/huggingface/lerobot/blob/main/src/lerobot/datasets/compute_stats.py).
 >
 > ```python
 > def _assert_type_and_shape(stats_list: list[dict[str, dict]]):

robomind2lerobot/robomind_h5.py

@@ -9,9 +9,9 @@ import numpy as np
 import pandas as pd
 import ray
 import torch
-from lerobot.common.datasets.compute_stats import aggregate_stats
-from lerobot.common.datasets.lerobot_dataset import LeRobotDataset, LeRobotDatasetMetadata
-from lerobot.common.datasets.utils import (
+from lerobot.datasets.compute_stats import aggregate_stats
+from lerobot.datasets.lerobot_dataset import LeRobotDataset, LeRobotDatasetMetadata
+from lerobot.datasets.utils import (
     check_timestamps_sync,
     get_episode_data_index,
     validate_episode_buffer,
@@ -20,7 +20,7 @@ from lerobot.common.datasets.utils import (
     write_episode_stats,
     write_info,
 )
-from lerobot.common.datasets.video_utils import get_safe_default_codec
+from lerobot.datasets.video_utils import get_safe_default_codec
 from ray.runtime_env import RuntimeEnv
 from robomind_uitls.configs import ROBOMIND_CONFIG
 from robomind_uitls.lerobot_uitls import compute_episode_stats, generate_features_from_config

robomind2lerobot/robomind_uitls/lerobot_uitls.py

@@ -1,7 +1,7 @@
 import numpy as np
 import torchvision
-from lerobot.common.datasets.compute_stats import auto_downsample_height_width, get_feature_stats, sample_indices
-from lerobot.common.datasets.utils import load_image_as_numpy
+from lerobot.datasets.compute_stats import auto_downsample_height_width, get_feature_stats, sample_indices
+from lerobot.datasets.utils import load_image_as_numpy
 
 torchvision.set_video_backend("pyav")