diff --git a/docs/demos/Video_Processing.py b/docs/demos/Video_Processing.py index 6d8e63b8e..31144aefb 100644 --- a/docs/demos/Video_Processing.py +++ b/docs/demos/Video_Processing.py @@ -469,12 +469,8 @@ def draw_detections(image, detections, show_class=False, show_score=False): prior_state = GaussianState(StateVector(np.zeros((6,1))), CovarianceMatrix(np.diag([100**2, 30**2, 100**2, 30**2, 100**2, 100**2]))) deleter_init = UpdateTimeStepsDeleter(time_steps_since_update=3) -initiator = MultiMeasurementInitiator(prior_state=prior_state, - deleter=deleter_init, - data_associator=data_associator, - updater=updater, - measurement_model=measurement_model, - min_points=10) +initiator = MultiMeasurementInitiator(prior_state, deleter_init, data_associator, updater, + measurement_model, min_points=10) # %% # Track Deletion diff --git a/stonesoup/initiator/simple.py b/stonesoup/initiator/simple.py index 88bb9945a..15d1a1020 100644 --- a/stonesoup/initiator/simple.py +++ b/stonesoup/initiator/simple.py @@ -7,7 +7,6 @@ from ..deleter import Deleter from ..models.base import LinearModel, ReversibleModel from ..models.measurement import MeasurementModel -from ..types.array import StateVector, CovarianceMatrix from ..types.hypothesis import SingleHypothesis from ..types.numeric import Probability from ..types.particle import Particle