Spaces:
Build error
Build error
Emil Ernerfeldt
commited on
Commit
·
f44e37f
1
Parent(s):
9901ded
Explain what is LeRobot-specific
Browse files
main.py
CHANGED
|
@@ -36,10 +36,10 @@ def to_rerun(column_name: str, value: Any) -> Any:
|
|
| 36 |
|
| 37 |
|
| 38 |
def log_dataset_to_rerun(dataset) -> None:
|
| 39 |
-
# Special time-like columns
|
| 40 |
TIME_LIKE = {"index", "frame_id", "timestamp"}
|
| 41 |
|
| 42 |
-
# Ignore these columns
|
| 43 |
IGNORE = {"episode_data_index_from", "episode_data_index_to", "episode_id"}
|
| 44 |
|
| 45 |
for row in tqdm(dataset):
|
|
|
|
| 36 |
|
| 37 |
|
| 38 |
def log_dataset_to_rerun(dataset) -> None:
|
| 39 |
+
# Special time-like columns for LeRobot datasets (https://huggingface.co/datasets/lerobot/):
|
| 40 |
TIME_LIKE = {"index", "frame_id", "timestamp"}
|
| 41 |
|
| 42 |
+
# Ignore these columns (again, LeRobot-specific):
|
| 43 |
IGNORE = {"episode_data_index_from", "episode_data_index_to", "episode_id"}
|
| 44 |
|
| 45 |
for row in tqdm(dataset):
|