fix: replace model_id in NVM functions

Author: zhenyuen

stonesoup/models/base_driver.py

@@ -263,7 +263,7 @@ def _jump_power(self, jsizes: np.ndarray) -> np.ndarray:
         return jsizes
 
     def _centering(
-        self, e_ft: np.ndarray, truncation: float, model_id: Optional[int] = None
+        self, e_ft: np.ndarray, truncation: float, mu_W: float
     ) -> StateVector:
         m = e_ft.shape[0]
         return np.zeros((m, 1))

stonesoup/models/driver.py

@@ -158,13 +158,13 @@ def _thinning_probabilities(self, jsizes: np.ndarray) -> np.ndarray:
         return (1.0 + self.beta * jsizes) * np.exp(-self.beta * jsizes)  # (n_jumps, n_samples)
 
     def _residual_covar(
-        self, e_ft: np.ndarray, truncation: float, model_id: int | None = None
+        self, e_ft: np.ndarray, truncation: float,  mu_W: float, sigma_W2: float
     ) -> CovarianceMatrix:
         m = e_ft.shape[0]
         return np.zeros((m, m))
 
     def _residual_mean(
-        self, e_ft: np.ndarray, truncation: float, model_id: int | None = None
+        self, e_ft: np.ndarray, truncation: float,  mu_W: float,
     ) -> CovarianceMatrix:
         m = e_ft.shape[0]
         return np.zeros((m, 1))

stonesoup/models/transition/levylinear.py

@@ -157,7 +157,7 @@ def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.A = self._init_A()
         self.h = self._init_h()
-
+    
     def _integrand(self, dt: float, jtimes: np.ndarray) -> np.ndarray:
         delta = dt - jtimes[..., np.newaxis, np.newaxis]
         return expm(self.A * delta) @ self.h