more dum

beneisner · beneisner · commit 46bba6ea62fb · 2024-05-16T01:21:20.000-04:00
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):
