Ensure variables get set in edge case scenario

micah-huth · micah-huth · commit 9655db2cc704 · 2021-03-25T14:22:57.000+10:00
diff --git a/roboticstoolbox/backends/VPython/grid.py b/roboticstoolbox/backends/VPython/grid.py
@@ -229,6 +229,11 @@ def __move_grid_objects(self):
                 y_origin = round(self.__scene.center.y, 2)
                 z_origin = 0
                 self.__focal_point = [x_origin, y_origin, z_origin]
+            else:
+                x_origin, y_origin, z_origin = round(self.__scene.center.x, 2), \
+                                               round(self.__scene.center.y, 2), \
+                                               round(self.__scene.center.z, 2)
+                self.__focal_point = [x_origin, y_origin, z_origin]
         else:
             x_origin, y_origin, z_origin = self.__focal_point[0], \
                                            self.__focal_point[1], \
