Fix problem generator for granite lab.

nasa · Feb 8, 2024 · e0db361 · e0db361
1 parent bc07049
commit e0db361
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/astrobee/survey_manager/survey_planner/src/survey_planner/problem_generator.py b/astrobee/survey_manager/survey_planner/src/survey_planner/problem_generator.py
@@ -468,7 +468,12 @@ def problem_generator(
     location_real_lines = [f"(location-real {bay})" for bay in bays]
     writer.declare_predicates(location_real_lines, "static_predicates")
 
-    candidates = (("jem_bay7", "berth1"), ("jem_bay7", "berth2"))
+    candidates = (
+        ("jem_bay7", "berth1"),
+        ("jem_bay7", "berth2"),
+        ("gra_bay1", "berth1"),
+        ("gra_bay3", "berth2"),
+    )
     dock_connected_lines = [
         f"(dock-connected {bay} {berth})"
         for bay, berth in candidates