fix: division by zero

thousandbrainsproject · Dec 11, 2024 · b197199 · b197199
1 parent 7db44ff
commit b197199
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/tbp/monty/frameworks/models/motor_policies.py b/src/tbp/monty/frameworks/models/motor_policies.py
@@ -1251,9 +1251,9 @@ def orienting_angle_from_normal(self, orienting: str) -> float:
         x, y, z = rotated_point_normal
 
         if "horizontal" == orienting:
-            return -np.degrees(np.arctan(x / z))
+            return -np.sign(x)*90.0 if z == 0 else -np.degrees(np.arctan(x / z))
         if "vertical" == orienting:
-            return -np.degrees(np.arctan(y / z))
+            return -np.sign(y)*90.0 if z == 0 else -np.degrees(np.arctan(y / z))
 
 
 ###