Small bug fix

bjmorgan · Aug 12, 2022 · c17fef5 · c17fef5
1 parent 16ad8a8
commit c17fef5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kinisi/diffusion.py b/kinisi/diffusion.py
@@ -323,7 +323,7 @@ def model_variance(dt: np.ndarray, a: float) -> np.ndarray:
         popt, _ = curve_fit(model_variance, self.dt[max_ngp:], self._v[max_ngp:])
         model_v = model_variance(self.dt[max_ngp:], *popt)
 
-        self._covariance_matrix = self.populate_covariance_matrix(model_v, self._n_i)
+        self._covariance_matrix = self.populate_covariance_matrix(model_v, self._n_i[max_ngp:])
         self._covariance_matrix = find_nearest_positive_definite(self._covariance_matrix)
 
         mv = multivariate_normal(self._n[max_ngp:], self._covariance_matrix, allow_singular=True)