more dum

r-pad · May 16, 2024 · 46bba6e · 46bba6e
1 parent e73245e
commit 46bba6e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/rpad/rlbench_utils/placement_dataset.py b/src/rpad/rlbench_utils/placement_dataset.py
@@ -436,8 +436,10 @@ def _select_action_vals(rgb, point_cloud, mask):
                 rgb,
                 point_cloud,
                 mask,
-                action_handles,
-                self.gripper_handles,
+                task_name=self.task_name,
+                phase=phase,
+                action_handles=action_handles,
+                gripper_handles=self.gripper_handles,
             )
 
         def _select_anchor_vals(rgb, point_cloud, mask):