Fix mypy error caused by new numpy typing

MichaelGrupp · Jan 28, 2025 · a342130 · a342130
1 parent 9fd2439
commit a342130
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/evo/tools/plot.py b/evo/tools/plot.py
@@ -596,7 +596,7 @@ def traj_xyz(axarr: np.ndarray, traj: trajectory.PosePath3D, style: str = '-',
             x = traj.timestamps
         xlabel = "$t$ (s)"
     else:
-        x = np.arange(0., len(traj.positions_xyz))
+        x = np.arange(0., len(traj.positions_xyz), dtype=float)
         xlabel = "index"
     ylabels = [
         f"$x$ ({length_unit.value})", f"$y$ ({length_unit.value})",
@@ -640,7 +640,7 @@ def traj_rpy(axarr: np.ndarray, traj: trajectory.PosePath3D, style: str = '-',
             x = traj.timestamps
         xlabel = "$t$ (s)"
     else:
-        x = np.arange(0., len(angles))
+        x = np.arange(0., len(angles), dtype=float)
         xlabel = "index"
     ylabels = ["$roll$ (deg)", "$pitch$ (deg)", "$yaw$ (deg)"]
     for i in range(0, 3):