From 7e3f284e672f6c152bdd76500ebac4e66252796d Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Sat, 20 Apr 2024 14:50:34 +0200 Subject: [PATCH] rebased from user/rcadene/2024_04_18_episode_data_index --- 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 1ae3493e..115cdec7 100644 --- a/lerobot/common/datasets/utils.py +++ b/lerobot/common/datasets/utils.py @@ -58,7 +58,7 @@ def hf_transform_to_torch(items_dict): def load_hf_dataset(dataset_id, version, root, split) -> datasets.Dataset: """hf_dataset contains all the observations, states, actions, rewards, etc.""" if root is not None: - hf_dataset = load_from_disk(Path(root) / dataset_id / split) + hf_dataset = load_from_disk(str(Path(root) / dataset_id / split)) else: # TODO(rcadene): remove dataset_id everywhere and use repo_id instead repo_id = f"lerobot/{dataset_id}"