From 302df104996eebe0eee5c36978596b0209803d73 Mon Sep 17 00:00:00 2001 From: Michel Aractingi Date: Mon, 2 Dec 2024 09:24:35 +0000 Subject: [PATCH] fixed bug in lang_instruction --- .../common/datasets/push_dataset_to_hub/openx_rlds_format.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/common/datasets/push_dataset_to_hub/openx_rlds_format.py b/lerobot/common/datasets/push_dataset_to_hub/openx_rlds_format.py index 27a94f5f5..1f8a5d144 100644 --- a/lerobot/common/datasets/push_dataset_to_hub/openx_rlds_format.py +++ b/lerobot/common/datasets/push_dataset_to_hub/openx_rlds_format.py @@ -202,7 +202,7 @@ def load_from_raw( # If lang_key is present, convert the entire tensor at once if lang_key is not None: - ep_dict["language_instruction"] = [str(x) for x in episode[lang_key]] + ep_dict["language_instruction"] = [x.numpy().decode("utf-8") for x in episode[lang_key]] ep_dict["timestamp"] = torch.arange(0, num_frames, 1) / fps ep_dict["episode_index"] = torch.tensor([ep_idx] * num_frames)