Add option to stop an experiment

2021-12-06 10:22:04 +01:00 · 2021-12-06 10:22:04 +01:00 · e5e17d743f
commit e5e17d743f
parent 55658948d6
1 changed files with 3 additions and 1 deletions
--- a/scripts/run.py
+++ b/scripts/run.py
@ -32,7 +32,9 @@ def main():
            controller.gripper.move(0.08)
            controller.switch_to_joint_trajectory_control()
            controller.moveit.goto("ready")
-            input("Press Enter to continue...")
+            i = input("Run policy? [y/n] ")
+            if i != "y":
+                exit()
            rospy.loginfo("Running policy ...")
        info = controller.run()
        logger.log_run(info)