From d52c6037e80d49a019af9285ae727027ead43831 Mon Sep 17 00:00:00 2001 From: Remi Cadene Date: Wed, 22 May 2024 09:03:43 +0000 Subject: [PATCH] fix --- .../common/datasets/push_dataset_to_hub/aloha_dora_format.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py b/lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py index 549afa89..6240860b 100644 --- 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)