diff --git a/pyroengine/engine.py b/pyroengine/engine.py index 845b7c1e..171be304 100644 --- a/pyroengine/engine.py +++ b/pyroengine/engine.py @@ -256,7 +256,7 @@ def _update_states(self, frame: Image.Image, preds: np.ndarray, cam_key: str) -> output_predictions = output_predictions[:5, :] # max 5 bbox self._states[cam_key]["last_predictions"].append( - (frame, preds, str(json.dumps(output_predictions.tolist())), datetime.utcnow().isoformat(), False) + (frame, preds, json.dumps(output_predictions.tolist()), datetime.utcnow().isoformat(), False) ) # update state @@ -390,7 +390,7 @@ def _process_alerts(self) -> None: lat=self.latitude, lon=self.longitude, media_id=self._alerts[0]["media_id"], - # localization=self._alerts[0]["localization"], + localization=self._alerts[0]["localization"], ) .json()["id"] )