tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Remi Cadene
2024-05-22 09:03:43 +00:00
parent b0cb342795
commit d52c6037e8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py
View File

@@ -93,7 +93,7 @@ def load_from_raw(raw_dir: Path, out_dir: Path):
# Remove rows with a NaN in any column. It can happened during the first frames of an episode, # Remove rows with a NaN in any column. It can happened during the first frames of an episode,
# because some cameras didnt start recording yet. # because some cameras didnt start recording yet.
data_df = data_df.dropna(axis=1) data_df = data_df.dropna(axis=0)
# Create symlink to raw videos directory (that needs to be absolute not relative) # Create symlink to raw videos directory (that needs to be absolute not relative)
# out_dir.mkdir(parents=True, exist_ok=True) # out_dir.mkdir(parents=True, exist_ok=True)