missed a step

r-pad · May 21, 2024 · 634f5b6 · 634f5b6
1 parent 70986d9
commit 634f5b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/rpad/rlbench_utils/placement_dataset.py b/src/rpad/rlbench_utils/placement_dataset.py
@@ -199,7 +199,7 @@ def get_anchor_points(
     if use_from_simulator:
         handle_mapping = {
             name: sim.simGetObjectHandle(name)
-            for name in BACKGROUND_NAMES + ROBOT_NONGRIPPER_NAMES
+            for name in BACKGROUND_NAMES + ROBOT_NONGRIPPER_NAMES + GRIPPER_OBJ_NAMES
         }
 
     if anchor_mode == AnchorMode.RAW: