fix

2024-05-22 09:03:43 +00:00
parent b0cb342795
commit d52c6037e8
1 changed files with 1 additions and 1 deletions
--- a/lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py
+++ b/lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py
@@ -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,
    # 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)
    # out_dir.mkdir(parents=True, exist_ok=True)