From a5ea759bc4a33d0c0aaf318cceb4978d52cb0b6f Mon Sep 17 00:00:00 2001 From: Pietro Date: Mon, 5 Apr 2021 09:12:10 +0200 Subject: [PATCH] Trajectory Result --- PyCTBN/PyCTBN/structure_graph/trajectory_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PyCTBN/PyCTBN/structure_graph/trajectory_generator.py b/PyCTBN/PyCTBN/structure_graph/trajectory_generator.py index 898ffa8..5b37706 100644 --- 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