Add unions to calc yaw and calc curvature

f1tenth · Jun 7, 2024 · d6f60e0 · d6f60e0
1 parent 597b0f8
commit d6f60e0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/f1tenth_gym/envs/track/cubic_spline.py b/f1tenth_gym/envs/track/cubic_spline.py
@@ -114,7 +114,7 @@ def calc_position(self, s: float) -> tuple[Union[float, None], Union[float, None
         """
         return self.spline(s)
 
-    def calc_curvature(self, s: float) -> float | None:
+    def calc_curvature(self, s: float) -> Union[float , None]:
         """
         Calc curvature at the given s.
 
@@ -134,7 +134,7 @@ def calc_curvature(self, s: float) -> float | None:
         k = (ddy * dx - ddx * dy) / ((dx**2 + dy**2) ** (3 / 2))
         return k
 
-    def calc_yaw(self, s: float) -> float | None:
+    def calc_yaw(self, s: float) -> Union[float, None]:
         """
         Calc yaw angle at the given s.