Move calculate_episode_data_index

2024-11-03 19:13:00 +01:00
parent 74270c8c91
commit 7b159a6b22
11 changed files with 71 additions and 66 deletions
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -7,8 +7,8 @@ import pytest
 import torch
 from datasets import Dataset

+from lerobot.common.datasets.push_dataset_to_hub.utils import calculate_episode_data_index
 from lerobot.common.datasets.utils import (
-    calculate_episode_data_index,
    hf_transform_to_torch,
 )
 from lerobot.common.utils.utils import (