dumb

r-pad · May 16, 2024 · bcb21cf · bcb21cf
1 parent 83e408d
commit bcb21cf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/rpad/rlbench_utils/placement_dataset.py b/src/rpad/rlbench_utils/placement_dataset.py
@@ -240,6 +240,8 @@ def get_action_points(
     rgb,
     point_cloud,
     mask,
+    task_name,
+    phase,
     use_from_simulator=False,
     action_handles=None,
     gripper_handles=None,