Fix #7

oseiskar · Sep 17, 2019 · bf89cbc · bf89cbc
1 parent fc37471
commit bf89cbc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/simdkalman/kalmanfilter.py b/simdkalman/kalmanfilter.py
@@ -519,7 +519,7 @@ def auto_flat_states(obs_gaussian):
             filtered_states = empty_gaussian(cov=True)
 
         if gains:
-            result.filtered.gains = np.empty((n_vars, n_measurements, n_states, n_states))
+            result.filtered.gains = np.empty((n_vars, n_measurements, n_states, n_obs))
 
         for j in range(n_measurements):
             if verbose: