From f70d54434602614fba10868d4e38f9a6b5f886b8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=9D=82=E6=9C=AC=E6=BB=89=E4=B9=9F?= Date: Tue, 15 Apr 2025 20:50:48 +0900 Subject: [PATCH] debug: change the data type of expected_shape to tuple --- lerobot/common/datasets/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/common/datasets/utils.py b/lerobot/common/datasets/utils.py index 9d8a54db..4b00fe0a 100644 --- a/lerobot/common/datasets/utils.py +++ b/lerobot/common/datasets/utils.py @@ -757,7 +757,7 @@ def validate_feature_numpy_array( if actual_dtype != np.dtype(expected_dtype): error_message += f"The feature '{name}' of dtype '{actual_dtype}' is not of the expected dtype '{expected_dtype}'.\n" - if actual_shape != expected_shape: + if actual_shape != tuple(expected_shape): error_message += f"The feature '{name}' of shape '{actual_shape}' does not have the expected shape '{expected_shape}'.\n" else: error_message += f"The feature '{name}' is not a 'np.ndarray'. Expected type is '{expected_dtype}', but type '{type(value)}' provided instead.\n"