fix the stack tasks

r-pad · May 28, 2024 · 66f8547 · 66f8547
1 parent 5d6167c
commit 66f8547
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/rpad/rlbench_utils/task_info.py b/src/rpad/rlbench_utils/task_info.py
@@ -289,7 +289,7 @@
 
 DOLLAR_STACK_OBJ_NAMES = ["dollar_stack"]
 SAFE_OBJ_NAMES = ["safe_body", "safe_door"]
-DOLLAR_STACK_POSE_NAME = "dollar_stack"
+DOLLAR_STACK_POSE_NAME = "dollar_front_visual"
 SAFE_POSE_NAME = "safe_body"
 
 PUT_MONEY_IN_SAFE = {
@@ -496,7 +496,8 @@
 SAFE_OBJ_NAMES = ["safe_body", "safe_door"]
 MONEY_OBJ_NAMES = ["dollar_stack0", "dollar_back_visual0"]
 SAFE_POSE_NAME = "safe_body"
-MONEY_POSE_NAME = "dollar_stack0"
+# MONEY_POSE_NAME = "dollar_stack0"
+MONEY_POSE_NAME = "dollar_front_visual0"
 
 
 TAKE_MONEY_OUT_SAFE = {