From 3d911acb34475fc26b41d6442178dcbf6b896cd2 Mon Sep 17 00:00:00 2001 From: Remi Date: Thu, 30 May 2024 11:01:07 +0200 Subject: [PATCH] Update lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py Co-authored-by: Thomas Wolf --- .../common/datasets/push_dataset_to_hub/aloha_dora_format.py | 3 ++- 1 file changed, 2 insertions(+), 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 b10e7b04..f3b83aa7 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 @@ -101,7 +101,8 @@ def load_from_raw(raw_dir: Path, out_dir: Path, fps: int): # sanity check has_nan = df.isna().any().any() - assert not has_nan + if has_nan: + raise ValueError("Dataset contains Nan values.") # sanity check episode indices go from 0 to n-1 ep_ids = [ep_idx for ep_idx, _ in df.groupby("episode_index")]