cup emptiness

bdaiinstitute · Feb 1, 2025 · d179bf6 · d179bf6
1 parent edd4815
commit d179bf6
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/predicators/ground_truth_models/mock_spot_env/nsrts.py b/predicators/ground_truth_models/mock_spot_env/nsrts.py
@@ -41,7 +41,7 @@ class MockSpotGroundTruthNSRTFactory(GroundTruthNSRTFactory):
 
     @classmethod
     def get_env_names(cls) -> Set[str]:
-        return {"mock_spot", "mock_spot_pick_place_two_cup", "mock_spot_drawer_cleaning",  "mock_spot_sort_weight"}
+        return {"mock_spot", "mock_spot_pick_place_two_cup", "mock_spot_drawer_cleaning",  "mock_spot_sort_weight", "mock_spot_cup_emptiness"}
 
     @classmethod
     def get_nsrts(cls, env_name: str, types: Dict[str, Type],

diff --git a/predicators/ground_truth_models/mock_spot_env/options.py b/predicators/ground_truth_models/mock_spot_env/options.py
@@ -43,7 +43,7 @@ class MockSpotGroundTruthOptionFactory(GroundTruthOptionFactory):
     @classmethod
     def get_env_names(cls) -> Set[str]:
         """Get the env names that this factory builds options for."""
-        return {"mock_spot", "mock_spot_pick_place_two_cup", "mock_spot_drawer_cleaning", "mock_spot_sort_weight"}
+        return {"mock_spot", "mock_spot_pick_place_two_cup", "mock_spot_drawer_cleaning", "mock_spot_sort_weight", "mock_spot_cup_emptiness"}
 
     @classmethod
     def get_options(cls, env_name: str, types: Dict[str, Type],

diff --git a/scripts/mock_experiments.py b/scripts/mock_experiments.py
@@ -158,7 +158,7 @@ def parse_args() -> argparse.Namespace:
     parser.add_argument("--planner", type=str,
                        help="Run only this planner (by name)")
     parser.add_argument('--env', type=str, default="mock_spot_drawer_cleaning", 
-                        choices=["mock_spot_drawer_cleaning", "mock_spot_pick_place_two_cup", "mock_spot_sort_weight"])
+                        choices=["mock_spot_drawer_cleaning", "mock_spot_pick_place_two_cup", "mock_spot_sort_weight", "mock_spot_cup_emptiness"])
     # parser.add_argument("--load_approach", action="store_true",
     #                    help="Load saved approach")
     return parser.parse_args()