fixing bug: array and keys did not match in particle_filter

nasa · Aug 17, 2023 · 6ace408 · 6ace408
1 parent b9e3ff5
commit 6ace408
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/progpy/state_estimators/particle_filter.py b/src/progpy/state_estimators/particle_filter.py
@@ -67,9 +67,8 @@ def __init__(self, model, x0, **kwargs):
                 # Added to avoid float/int issues
                 self.parameters['num_particles'] = int(self.parameters['num_particles'])
             sample_gen = x0.sample(self.parameters['num_particles'])
-        samples = [array(sample_gen.key(k), dtype=float64) for k in x0.keys()]
-
-        self.particles = model.StateContainer(array(samples, dtype=float64))
+        samples = {k: array(sample_gen.key(k), dtype=float64) for k in x0.keys()}
+        self.particles = model.StateContainer(samples)
 
         if 'R' in self.parameters:
             # For backwards compatibility