Update to only permute if observations are gaussian

bonsai-rx · Jul 10, 2024 · a3f54a1 · a3f54a1
1 parent d13cb16
commit a3f54a1
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/Bonsai.ML.HiddenMarkovModels/main.py b/src/Bonsai.ML.HiddenMarkovModels/main.py
@@ -150,9 +150,11 @@ def start_loop(loop):
                     loop.run_forever()
 
                 def on_completion(future):
-                    permutation = calculate_permutation(
-                        self.observation_params[0], self.params[2][0])
-                    super(HiddenMarkovModel, self).permute(permutation)
+
+                    if self.observation_type == "gaussian":
+                        permutation = calculate_permutation(
+                            self.observation_params[0], self.params[2][0])
+                        super(HiddenMarkovModel, self).permute(permutation)
 
                     initial_state_distribution = None if vars_to_estimate[
                         "initial_state_distribution"] else self.initial_state_distribution