refactor(dataset): update imports across the codebase

This commit is contained in:
Steven Palma
2026-03-15 23:09:52 -07:00
parent 26d732c8c8
commit d0ae3e9481
49 changed files with 342 additions and 410 deletions
+7 -5
View File
@@ -20,17 +20,19 @@ import pandas as pd
import pytest
from datasets import Dataset
from lerobot.datasets.utils import (
DEFAULT_CHUNK_SIZE,
DEFAULT_DATA_FILE_SIZE_IN_MB,
DEFAULT_DATA_PATH,
from lerobot.datasets.io_utils import (
get_hf_dataset_size_in_mb,
update_chunk_file_indices,
write_episodes,
write_info,
write_stats,
write_tasks,
)
from lerobot.datasets.utils import (
DEFAULT_CHUNK_SIZE,
DEFAULT_DATA_FILE_SIZE_IN_MB,
DEFAULT_DATA_PATH,
update_chunk_file_indices,
)
def write_hf_dataset(