Remove mask check key in XarmProcessor (#117)

2024-04-29 22:00:37 +02:00
parent 508bd92d03
commit 1ec5f77f7c
1 changed files with 1 additions and 1 deletions
--- a/lerobot/common/datasets/push_dataset_to_hub/xarm_processor.py
+++ b/lerobot/common/datasets/push_dataset_to_hub/xarm_processor.py
@@ -18,7 +18,7 @@ class XarmProcessor:

    def __init__(self, folder_path: str, fps: int | None = None):
        self.folder_path = Path(folder_path)
-        self.keys = {"actions", "rewards", "dones", "masks"}
+        self.keys = {"actions", "rewards", "dones"}
        self.nested_keys = {"observations": {"rgb", "state"}, "next_observations": {"rgb", "state"}}
        if fps is None:
            fps = 15