Merge pull request #116 from decargroup/issue_114_imm_not_workinmg

Bugfix IMM
decargroup · Feb 27, 2024 · 9611b74 · 9611b74
2 parents 7a54c8a + 159be95
commit 9611b74
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/ex_imm_se3.py b/examples/ex_imm_se3.py
@@ -189,7 +189,7 @@ def ekf_trial(trial_number: int) -> List[nav.GaussianResult]:
         average_model_probabilities = np.average(
             np.array([t.model_probabilities for t in results.trial_results]),
             axis=0,
-        )
+        ).T
         fig, ax = plt.subplots(1, 1)
         for lv1 in range(n_models):
             ax.plot(results.stamp, average_model_probabilities[lv1, :])