fix pick up task

bdaiinstitute · Nov 13, 2024 · 44cd027 · 44cd027
1 parent 95442a5
commit 44cd027
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/predicators/envs/spot_env.py b/predicators/envs/spot_env.py
@@ -3533,10 +3533,11 @@ def _detection_id_to_obj(self) -> Dict[ObjectDetectionID, Object]:
             ("cardboard_box", "cardboard_box", _container_type),
             # ("block", "red_block", _movable_object_type),
             # DEBUG objects:
-            # ("block", "orange_cup", _container_type),
-            # ("block", "spam_box", _container_type), 
+            # NOTE: should use movable type, because the goal for this task assumes movable
+            # ("block", "orange_cup", _movable_object_type),
+            # ("block", "spam_box", _movable_object_type), 
             # ("block", "yellow_apple", _movable_object_type),
-            ("block", "green_cup", _container_type),
+            ("block", "green_cup", _movable_object_type),
             # ("block", "green_block", _movable_object_type),
             # ("block", "green_apple", _movable_object_type),
         ]