From 46bba6ea62fbacc14a359186ee4c8adf1075a17a Mon Sep 17 00:00:00 2001
From: Ben Eisner <ben.a.eisner@gmail.com>
Date: Thu, 16 May 2024 01:21:20 -0400
Subject: [PATCH] more dum

---
 src/rpad/rlbench_utils/placement_dataset.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/rpad/rlbench_utils/placement_dataset.py b/src/rpad/rlbench_utils/placement_dataset.py
index 6eec5da..a33c3bd 100644
--- 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):