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_sampler.py
+++ b/tests/test_sampler.py
@@ -15,9 +15,9 @@
 # limitations under the License.
 from datasets import Dataset

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