Merge pull request #283 from chris-konrad/trajectory-plot-timesteps

Correcting time step width of trajectory plots to equal node_time_int…
csu-hmc · Dec 20, 2024 · cbea1dc · cbea1dc
2 parents aae009e + bca3d41
commit cbea1dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/opty/direct_collocation.py b/opty/direct_collocation.py
@@ -397,7 +397,7 @@ def plot_trajectories(self, vector, axes=None):
             node_time_interval = self.collocator.node_time_interval
 
         time = np.linspace(0,
-                           self.collocator.num_collocation_nodes *
+                           (self.collocator.num_collocation_nodes-1) *
                            node_time_interval,
                            num=self.collocator.num_collocation_nodes)