Fix planner script.

nasa · Jan 24, 2024 · 93eed3c · 93eed3c
1 parent 3fc574f
commit 93eed3c
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/astrobee/survey_manager/survey_planner/CMakeLists.txt b/astrobee/survey_manager/survey_planner/CMakeLists.txt
@@ -75,6 +75,7 @@ catkin_install_python(PROGRAMS
   tools/command_astrobee
   tools/monitor_astrobee
   tools/plan_interpreter
+  tools/plan_survey
   tools/problem_generator
   DESTINATION ${CATKIN_PACKAGE_BIN_DESTINATION})
 

diff --git a/astrobee/survey_manager/survey_planner/src/survey_planner/plan_survey.py b/astrobee/survey_manager/survey_planner/src/survey_planner/plan_survey.py
@@ -29,7 +29,7 @@
 
 import pyparsing as pp
 import yaml
-from problem_generator import PDDL_DIR
+from survey_planner.problem_generator import PDDL_DIR
 
 LocationName = str  # Names PDDL object of type location
 LocationIndex = int  # Index of location in CONFIG.locations

diff --git a/astrobee/survey_manager/survey_planner/tools/plan_survey b/astrobee/survey_manager/survey_planner/tools/plan_survey
@@ -20,7 +20,7 @@
 
 import sys
 
-from survey_planner import survey_planner
+from survey_planner import plan_survey
 
 if __name__ == "__main__":
-    sys.exit(survey_planner.main())
+    sys.exit(plan_survey.main())