Spaces:
Sleeping
Sleeping
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):
|