fix mypy issue

open-dynamic-robot-initiative · Sep 21, 2023 · 8b5407f · 8b5407f
1 parent cba6b02
commit 8b5407f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/robot_fingers/utils.py b/robot_fingers/utils.py
@@ -1,5 +1,6 @@
 """Utility classes/functions for the robot_fingers package."""
 import time
+import typing as t
 
 import numpy as np
 import numpy.typing as npt
@@ -36,7 +37,7 @@ def update(self) -> None:
 
 def min_jerk_trajectory(
     start_position: npt.ArrayLike, end_position: npt.ArrayLike, num_steps: int
-) -> npt.NDArray:
+) -> t.Iterator[npt.NDArray]:
     """Generator for computing minimum jerk trajectories.
 
     Example: