wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix vizualize

This commit is contained in:
Simon Alibert 2024-11-03 18:18:23 +01:00
parent 16103cb8b7
commit 95a4b59b5b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/scripts/visualize_dataset_html.py
View File

@ -136,10 +136,10 @@ def write_episode_data_csv(output_dir, file_name, episode_index, dataset):
# init header of csv with state and action names # init header of csv with state and action names
header = ["timestamp"] header = ["timestamp"]
if has_state: if has_state:
dim_state = dataset.shapes["observation.state"] dim_state = dataset.meta.shapes["observation.state"]
header += [f"state_{i}" for i in range(dim_state)] header += [f"state_{i}" for i in range(dim_state)]
if has_action: if has_action:
dim_action = dataset.shapes["action"] dim_action = dataset.meta.shapes["action"]
header += [f"action_{i}" for i in range(dim_action)] header += [f"action_{i}" for i in range(dim_action)]
columns = ["timestamp"] columns = ["timestamp"]