Skip to content

Commit 6ace408

Browse files
committed
fixing bug: array and keys did not match in particle_filter
1 parent b9e3ff5 commit 6ace408

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

src/progpy/state_estimators/particle_filter.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -67,9 +67,8 @@ def __init__(self, model, x0, **kwargs):
6767
# Added to avoid float/int issues
6868
self.parameters['num_particles'] = int(self.parameters['num_particles'])
6969
sample_gen = x0.sample(self.parameters['num_particles'])
70-
samples = [array(sample_gen.key(k), dtype=float64) for k in x0.keys()]
71-
72-
self.particles = model.StateContainer(array(samples, dtype=float64))
70+
samples = {k: array(sample_gen.key(k), dtype=float64) for k in x0.keys()}
71+
self.particles = model.StateContainer(samples)
7372

7473
if 'R' in self.parameters:
7574
# For backwards compatibility

0 commit comments

Comments
 (0)