Do not save waypoints if show_movement has been disabled

urastogi885 · Mar 9, 2021 · bd411b2 · bd411b2
1 parent b191988
commit bd411b2
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/utils/planner.py b/utils/planner.py
@@ -101,7 +101,8 @@ def explore(self, show_movement=False):
             if self.robot == self.target:
                 print('Reached Target!!...')
                 self.video_output.release()
-                np.save('waypoints.npy', waypoints)
-                for key in self.grid_obj.obstacles.keys():
-                    np.save(key, self.grid_obj.obstacles[key])
+                if show_movement:
+                    np.save('waypoints.npy', waypoints)
+                    for key in self.grid_obj.obstacles.keys():
+                        np.save(key, self.grid_obj.obstacles[key])
                 break