From 1ec5f77f7c5125870f750188ab2c0371398bdb07 Mon Sep 17 00:00:00 2001
From: Adil Zouitine <adilzouitinegm@gmail.com>
Date: Mon, 29 Apr 2024 22:00:37 +0200
Subject: [PATCH] Remove mask check key in XarmProcessor (#117)

---
 lerobot/common/datasets/push_dataset_to_hub/xarm_processor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lerobot/common/datasets/push_dataset_to_hub/xarm_processor.py b/lerobot/common/datasets/push_dataset_to_hub/xarm_processor.py
index 57401955..541d83e8 100644
--- 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