Emil Ernerfeldt commited on
Commit
f44e37f
1 Parent(s): 9901ded

Explain what is LeRobot-specific

Browse files
Files changed (1) hide show
  1. main.py +2 -2
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):