admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-21 19:49:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix hf_dataset.set_transform(hf_transform_to_torch)

Browse Source
This commit is contained in:
Remi Cadene
2025-04-23 11:42:21 +02:00
committed by Michel Aractingi
parent 01bc89b6f4
commit ad1ad11eac
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/datasets/test_delta_timestamps.py
+2 -2
View File
@@ -56,8 +56,8 @@ def calculate_episode_data_index(hf_dataset: datasets.Dataset) -> dict[str, np.n
def synced_timestamps_factory(hf_dataset_factory): def synced_timestamps_factory(hf_dataset_factory):
def _create_synced_timestamps(fps: int = 30) -> tuple[np.ndarray, np.ndarray, np.ndarray]: def _create_synced_timestamps(fps: int = 30) -> tuple[np.ndarray, np.ndarray, np.ndarray]:
hf_dataset = hf_dataset_factory(fps=fps) hf_dataset = hf_dataset_factory(fps=fps)
timestamps = hf_dataset["timestamp"].numpy() timestamps = torch.stack(hf_dataset["timestamp"]).numpy()
episode_indices = hf_dataset["episode_index"].numpy() episode_indices = torch.stack(hf_dataset["episode_index"]).numpy()
episode_data_index = calculate_episode_data_index(hf_dataset) episode_data_index = calculate_episode_data_index(hf_dataset)
return timestamps, episode_indices, episode_data_index return timestamps, episode_indices, episode_data_index