Trajectory Result

4 years ago · a5ea759bc4
parent e000967852
commit a5ea759bc4
1 changed files with 1 additions and 1 deletions
--- a/PyCTBN/PyCTBN/structure_graph/trajectory_generator.py
+++ b/PyCTBN/PyCTBN/structure_graph/trajectory_generator.py
@ -54,7 +54,7 @@ class TrajectoryGenerator(object):
            t = time[next]

            if t >= t_end:
-                return sigma
+                return Trajectory(self._importer.build_list_of_samples_array(sigma), len(self._vnames) + 1)
            else:
                cim_row = np.array(cim[current_values.at[self._vnames[next]]])
                cim_row[current_values.at[self._vnames[next]]] = 0